Update app.py

app.py

@@ -41,15 +41,16 @@ 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= pred_ids[0].cpu().detach()
+        #pred_ids= pred_ids[0].cpu().detach()
     return pred_ids
         #xcp = processor.decode(pred_ids)
         #return xcp
 
 def dec(pids):
     with torch.inference_mode():
-        xcp = processor.decode(pids)
+        xcp = processor.batch_decode(pids)
     return xcp
+    # decode torch.cat( pids ??
 
 bl = gr.Blocks()
 with bl: