amandakonet commited on
Commit
a1beca9
1 Parent(s): a346039

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = ['refutes', 'supports', 'not enough info']
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)