Update app.py
Browse files
app.py
CHANGED
@@ -41,7 +41,7 @@ def recc(a_f):
|
|
41 |
input_values = torch.tensor(input_values, device=device).unsqueeze(0)
|
42 |
logits = model(input_values).logits
|
43 |
pred_ids = torch.argmax(logits, dim=-1)
|
44 |
-
|
45 |
return pred_ids
|
46 |
#xcp = processor.decode(pred_ids)
|
47 |
#return xcp
|
|
|
41 |
input_values = torch.tensor(input_values, device=device).unsqueeze(0)
|
42 |
logits = model(input_values).logits
|
43 |
pred_ids = torch.argmax(logits, dim=-1)
|
44 |
+
pred_ids= list(pred_ids[0].cpu().detach())
|
45 |
return pred_ids
|
46 |
#xcp = processor.decode(pred_ids)
|
47 |
#return xcp
|