snaramirez872 commited on
Commit
ef7bc7f
·
1 Parent(s): e55d629

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -126,7 +126,7 @@ def predict(tweets):
126
  mod.eval()
127
  res = []
128
  with torch.no_grad():
129
- for ins in testing_loader:
130
  outs = mod(input_ids=ins['input_ids'].to(device), attention_mask=ins['attention_mask'].to(device), token_type_ids=ins['token_type_ids'].to(device))
131
  probs = torch.softmax(outs[0], dim=-1)
132
  preds = torch.argmax(probs, dim=-1)
 
126
  mod.eval()
127
  res = []
128
  with torch.no_grad():
129
+ for ins in tweets:
130
  outs = mod(input_ids=ins['input_ids'].to(device), attention_mask=ins['attention_mask'].to(device), token_type_ids=ins['token_type_ids'].to(device))
131
  probs = torch.softmax(outs[0], dim=-1)
132
  preds = torch.argmax(probs, dim=-1)