Update app.py

app.py

@@ -3,11 +3,9 @@ from transformers import pipeline
 import gradio as gr
 import torch
 
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 asr = pipeline(model="asif00/whisper-bangla")
-# asr.to(device=device)
 ser = pipeline("text2text-generation", model="asif00/mbart_bn_error_correction")
-ser.to(device=device)
+
 
 @spaces.GPU
 def transcribe(audio):
@@ -18,9 +16,9 @@ def transcribe(audio):
 
 @spaces.GPU
 def correction(text):
-    corrected_text = ser(text)["generated_text"]
-    print(corrected_text)
-    return corrected_text
+    corrected_text = ser(text)
+    print(corrected_text["generated_text"])
+    return corrected_text["generated_text"]
 
 
 def transcribe_and_correct(audio):