test

app.py

@@ -29,18 +29,15 @@ def convert_to_mp3(video_file):
 @st.cache_resource
 def get_model():
     model_size = "large-v3"
-    os.environ["KMP_DUPLICATE_LIB_OK"] = "TRUE"
-    model = WhisperModel(model_size, device="auto", compute_type="float16")
+    model = WhisperModel(model_size)
     return model
 
 
 def transcribe_audio(audio_file):
     model = get_model()
-    segments, info = model.transcribe(audio_file, beam_size=5)
+    segments, info = model.transcribe(audio_file)
     return segments
 
-
-# C6L0hxvouRy
 def transcribe_post():
     file_name = None
     media_pk: str = None
@@ -73,9 +70,13 @@ def transcribe_post():
 
             file_transcript = ""
             with st.spinner("Final step: transcribing audio..."):
-                segments = transcribe_audio(file_name_audio)
-                for segment in segments:
-                    file_transcript += "[" + str(segment.start) + "s -> " + str(segment.end) + "s]" + segment.text + "\r\n\r\n"
+                try:
+                    segments = transcribe_audio(file_name_audio)
+                    for segment in segments:
+                        file_transcript += "[" + str(segment.start) + "s -> " + str(segment.end) + "s]" + segment.text + "\r\n\r\n"
+                except Exception as e:
+                    st.error(traceback.format_exc())
+                    st.error("Cannot transcribe audio!")
             if not file_transcript:
                 st.error("No transcription found!")
             else: