updated app.py

app.py

@@ -4,7 +4,7 @@ import gradio as gr
 import os
 
 #%%
-whisper = pipeline(model='C:\\Users\\doornj\\Desktop\\WhisperANSP\\Models\\whisper-large-v2-atco2-asr-atcosim-ANSP-3h1m', task='automatic-speech-recognition')
+whisper = pipeline(model='/mnt/projects/whisper/WhisperANSP/Models/whisper-large-v2-atco2-asr-atcosim-ANSP-3h1m', task='automatic-speech-recognition')
 bert_atco_ner = pipeline(model='Jzuluaga/bert-base-ner-atc-en-atco2-1h')
 
 #%%
@@ -36,8 +36,8 @@ def extractCallSignCommand(transcription):
         return 'There was no transcription to extract a callsign or command from...'
 
 #%%
-def transcribeAndExtract(audio_mic, audio_file, transcribe_only):
-    transcription = transcribe(audio_mic, audio_file)
+def transcribeAndExtract(audio_file, audio_mic, transcribe_only):
+    transcription = transcribe(audio_file, audio_mic)
     if not transcribe_only:
         callSignCommandValues = extractCallSignCommand(transcription)
     else:
@@ -47,11 +47,11 @@ def transcribeAndExtract(audio_mic, audio_file, transcribe_only):
 #%%
 iface = gr.Interface(
         fn=transcribeAndExtract,
-        inputs=[gr.Audio(source='upload', type='filepath'), gr.Audio(source='microphone', type='filepath'), gr.Checkbox(label='Transcribe only', default=False)],
+        inputs=[gr.Audio(source='upload', type='filepath', interactive=True), gr.Audio(source='microphone', type='filepath'), gr.Checkbox(label='Transcribe only', default=False)],
         outputs=[gr.Text(label='Transcription'), gr.Text(label='Callsigns, commands and values')],
         title='Whisper Large v2 - ATCO2-ATCOSIM-ANSP',
         description='This demo will transcribe ATC audio files by using the Whisper Large v2 model fine-tuned on the ATCO2, ATCOSIM and ANSP datasets. Further it uses a Named Entity Recognition model to extract callsigns, commands and values from the transcription. This model is based on Google\'s BERT model and fine-tuned on the ATCO2 dataset.',
 )
 
 #%%
-iface.launch()
+iface.launch(server_name='0.0.0.0', server_port=9000)