Spaces:
Runtime error
Runtime error
snaramirez872
commited on
Commit
·
ef7bc7f
1
Parent(s):
e55d629
Update app.py
Browse files
app.py
CHANGED
@@ -126,7 +126,7 @@ def predict(tweets):
|
|
126 |
mod.eval()
|
127 |
res = []
|
128 |
with torch.no_grad():
|
129 |
-
for ins in
|
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)
|