Spaces:
Runtime error
Runtime error
amandakonet
commited on
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)
|