Update app.py

app.py

@@ -41,8 +41,9 @@ 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)[0]
-        xcp = processor.decode(pred_ids)
-        return xcp
+        return pred_ids
+        #xcp = processor.decode(pred_ids)
+        #return xcp
     
 
 bl = gr.Blocks()