Update app.py

app.py

@@ -5,19 +5,20 @@ import torch
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 asr = pipeline(model="asif00/whisper-bangla")
-asr.to(device=device)
+# asr.to(device=device)
 ser = pipeline("text2text-generation", model="asif00/mbart_bn_error_correction")
 ser.to(device=device)
 
 @spaces.GPU
 def transcribe(audio):
     text = asr(audio)["text"]
+    print(text)
     return text
 
 
 @spaces.GPU
 def correction(text):
-    corrected_text = ser(text)
+    corrected_text = ser(text)["generated_text"]
     print(corrected_text)
     return corrected_text