Update app.py

app.py

@@ -40,7 +40,8 @@ def recc(a_f):
         input_values = processor(wav,sampling_rate=16000).input_values[0]
         input_values = torch.tensor(input_values, device=device).unsqueeze(0)
         logits = model(input_values).logits
-        pred_ids = torch.argmax(logits, dim=-1)[0] #s[0].cpu().detach()
+        pred_ids = torch.argmax(logits, dim=-1)
+        pred_ids= pred_ids[0].cpu().detach()
     return pred_ids
         #xcp = processor.decode(pred_ids)
         #return xcp