Update app.py

app.py

@@ -16,8 +16,9 @@ processor_is = Wav2Vec2Processor.from_pretrained(MODEL_IS)
 model_fo = Wav2Vec2ForCTC.from_pretrained(MODEL_FO).to(device)
 processor_fo = Wav2Vec2Processor.from_pretrained(MODEL_FO) 
 
-
 pipe_is = pipeline(model=MODEL_IS)
+pipe_fo = pipeline(model=MODEL_FO)
+
 
 
 def readwav(a_f):
@@ -42,11 +43,13 @@ def recc(audio_file,model,processor):
     
 def recis(audio_file):
     single_output = recc(audio_file,model_is,processor_is)
-    chunk_output = pipe_is(audio_file, chunk_length_s=6)['text']
+    chunk_output = pipe_is(audio_file, chunk_length_s=4)['text']
     return (single_output, chunk_output)
 
 def recfo(audio_file):
-    return recc(audio_file,model_fo,processor_fo)
+    single_output = recc(audio_file,model_fo,processor_fo)
+    chunk_output = pipe_fo(audio_file, chunk_length_s=4)['text']
+    return (single_output, chunk_output)
 
 
 bl = gr.Blocks()
@@ -71,15 +74,17 @@ with bl:
                 audio_file = gr.Audio(type="filepath")
                 with gr.Column():
                     whole_output = gr.Textbox(label="whole-file recognition")
-                    chunk_output = gr.Textbox("recognition with chunking")
+                    chunk_output = gr.Textbox(label="recognition with chunking")
             text_button = gr.Button("Recognise Icelandic")
             text_button.click(recis, inputs=audio_file, outputs=[whole_output,chunk_output])
         with gr.TabItem("Faroese"):
             with gr.Row():
                 audio_file = gr.Audio(type="filepath")
-                text_output = gr.Textbox()
+                with gr.Column():
+                    whole_output = gr.Textbox(label="whole-file recognition")
+                    chunk_output = gr.Textbox(label="recognition with chunking")
             text_button = gr.Button("Recognise Faroese")
-            text_button.click(recfo, inputs=audio_file, outputs=text_output)
+            text_button.click(recfo, inputs=audio_file, outputs=[whole_output,chunk_output])
 
 bl.launch()