Update app.py

app.py

@@ -40,7 +40,7 @@ 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)
+        pred_ids = torch.argmax(logits, dim=-1)[0]
         xcp = processor.decode(pred_ids)
         return xcp