Update app.py

app.py

@@ -41,15 +41,10 @@ def recc(a_f):
         input_values = torch.tensor(input_values, device=device).unsqueeze(0)
         logits = model(input_values).logits
         pred_ids = torch.argmax(logits, dim=-1)
-        pred_ids= list(pred_ids[0].cpu().detach())
-    return pred_ids
-        #xcp = processor.decode(pred_ids)
-        #return xcp
+        #pred_ids= pred_ids[0].cpu().detach()
+        xcp = processor.batch_decode(pred_ids)
+        return xcp
 
-def dec(pids):
-    xccp = processor.decode(pids)
-    return xccp
-    # decode torch.cat( pids ??
 
 bl = gr.Blocks()
 with bl:
@@ -59,10 +54,6 @@ with bl:
     
     text_button.click(recc, inputs=audio_file, outputs=text_output)
 
-    text_button2 = gr.Button("Dec")
-    text_output2 = gr.Textbox()
-    text_button2.click(dec, inputs=text_output, outputs=text_output2)
-
 
 bl.launch()