xuandin commited on
Commit
15a70bc
·
verified ·
1 Parent(s): be06814

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -91,6 +91,7 @@ def perform_verification(claim, context, model_qatc, tokenizer_qatc, model_tc, t
91
  # Classify the claim
92
  verdict_start_time = time.time()
93
  with torch.no_grad():
 
94
  prob3class, pred_tc = classify_claim(
95
  claim, evidence, model_tc, tokenizer_tc, DEVICE
96
  )
@@ -102,8 +103,6 @@ def perform_verification(claim, context, model_qatc, tokenizer_qatc, model_tc, t
102
  claim, evidence, model_bc, tokenizer_bc, DEVICE
103
  )
104
  verdict = "SUPPORTED" if pred_bc == 0 else "REFUTED" if prob2class > prob3class else ["NEI", "SUPPORTED", "REFUTED"][pred_tc]
105
- else:
106
- verdict = "NEI"
107
 
108
  verdict_time = time.time() - verdict_start_time
109
 
 
91
  # Classify the claim
92
  verdict_start_time = time.time()
93
  with torch.no_grad():
94
+ verdict = "NEI"
95
  prob3class, pred_tc = classify_claim(
96
  claim, evidence, model_tc, tokenizer_tc, DEVICE
97
  )
 
103
  claim, evidence, model_bc, tokenizer_bc, DEVICE
104
  )
105
  verdict = "SUPPORTED" if pred_bc == 0 else "REFUTED" if prob2class > prob3class else ["NEI", "SUPPORTED", "REFUTED"][pred_tc]
 
 
106
 
107
  verdict_time = time.time() - verdict_start_time
108