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 |
|