Update main.py

main.py

@@ -20,12 +20,15 @@ transcribe_token_id = all_special_ids[-5]
 translate_token_id = all_special_ids[-6]
 
 def transcribe(file):
-    pipe.model.config.forced_decoder_ids = [[2, transcribe_token_id]]
-    text = pipe(file, return_timestamps=True)
-    file_path = '/home/user/data/new_file.txt'
-    with open(file_path, "w") as file:
-        file.write(text)
-    return text
+    try:
+        pipe.model.config.forced_decoder_ids = [[2, transcribe_token_id]]
+        text = pipe(file, return_timestamps=True)
+        file_path = '/home/user/data/new_file.txt'
+        with open(file_path, "w") as file:
+            file.write(text)
+        return text
+    except error:
+        print(error)
 
 app = FastAPI()