Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
|