Spaces:
Runtime error
Runtime error
Commit
·
a1beca9
1
Parent(s):
a346039
Update app.py
Browse files
app.py
CHANGED
|
@@ -202,7 +202,7 @@ features = tokenizer(option_claim, option_evidence,
|
|
| 202 |
model.eval()
|
| 203 |
with torch.no_grad():
|
| 204 |
scores = model(**features).logits
|
| 205 |
-
label_mapping = ['
|
| 206 |
labels = [label_mapping[score_max] for score_max in scores.argmax(dim=1)]
|
| 207 |
|
| 208 |
st.write("**Claim:**", option_claim)
|
|
|
|
| 202 |
model.eval()
|
| 203 |
with torch.no_grad():
|
| 204 |
scores = model(**features).logits
|
| 205 |
+
label_mapping = ['supports', 'refutes', 'not enough info']
|
| 206 |
labels = [label_mapping[score_max] for score_max in scores.argmax(dim=1)]
|
| 207 |
|
| 208 |
st.write("**Claim:**", option_claim)
|