Update app.py

app.py

@@ -158,14 +158,11 @@ mms_model.load_adapter("mkd")
 
 # Create a partial function with the device pre-applied
 return_prediction_whisper_with_device = partial(return_prediction_whisper, device=device)
-return_prediction_w2v2_with_device = partial(return_prediction_w2v2, device=device)
+# return_prediction_w2v2_with_device = partial(return_prediction_w2v2, device=device)
 return_prediction_with_device_compare = partial(return_prediction_compare, device=device)
 
 
 # Load the ASR models
-w2v2_classifier = foreign_class(source="Macedonian-ASR/wav2vec2-aed-macedonian-asr", pymodule_file="custom_interface_app.py", classname="ASR")
-w2v2_classifier = w2v2_classifier.to(device)
-w2v2_classifier.eval()
 whisper_classifier = foreign_class(source="Macedonian-ASR/whisper-large-v3-macedonian-asr", pymodule_file="custom_interface_app.py", classname="ASR")
 whisper_classifier = whisper_classifier.to(device)
 whisper_classifier.eval()
@@ -179,38 +176,6 @@ recap_model.to(device)
 recap_model.eval()
 
 
-mic_transcribe_whisper = gr.Interface(
-    fn=return_prediction_whisper_with_device,
-    inputs=gr.Audio(sources="microphone", type="filepath"),
-    outputs=gr.Textbox(),
-    allow_flagging="never",
-    live=False,
-)
-
-# file_transcribe_whisper = gr.Interface(
-#     fn=return_prediction_whisper_with_device,
-#     inputs=gr.Audio(sources="upload", type="filepath"),
-#     outputs=gr.Textbox(),
-#     allow_flagging="never",
-#     live=False
-# )
-
-mic_transcribe_w2v2 = gr.Interface(
-    fn=return_prediction_w2v2_with_device,
-    inputs=gr.Audio(sources="microphone", type="filepath"),
-    outputs=gr.Textbox(),
-    allow_flagging="never",
-    live=False,
-)
-
-
-# file_transcribe_w2v2 = gr.Interface(
-#     fn=return_prediction_w2v2_with_device,
-#     inputs=gr.Audio(sources="upload", type="filepath"),
-#     outputs=gr.Textbox(),
-#     allow_flagging="never",
-#     live=False
-# )
 
 mic_transcribe_compare = gr.Interface(
     fn=return_prediction_with_device_compare,
@@ -271,8 +236,8 @@ with transcriber_app:
     # state = gr.State(value=[], delete_callback=lambda v: print("STATE DELETED"))
 
     gr.TabbedInterface(
-        [mic_transcribe_whisper, mic_transcribe_w2v2, mic_transcribe_compare],
-        ["Буки-Whisper транскрипција", "Буки-W2v2 транскрипција", "Споредба на модели"],
+        [mic_transcribe_compare],
+        ["Споредба на модели"],
     )
     state = gr.State(value=[], delete_callback=lambda v: print("STATE DELETED"))