clr commited on
Commit
9bf7dee
·
1 Parent(s): d884e8f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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
- #pred_ids= pred_ids[0].cpu().detach()
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