Update app.py
Browse files
app.py
CHANGED
|
@@ -82,7 +82,7 @@ def classify_news(text):
|
|
| 82 |
pred1 = l1_model.config.id2label[probs1.argmax().item()]
|
| 83 |
|
| 84 |
if conf1 < CONFIDENCE_THRESHOLD:
|
| 85 |
-
return "Other", f"L1(
|
| 86 |
|
| 87 |
l2_tok, l2_mod = l2_manager.get_model(pred1)
|
| 88 |
|
|
|
|
| 82 |
pred1 = l1_model.config.id2label[probs1.argmax().item()]
|
| 83 |
|
| 84 |
if conf1 < CONFIDENCE_THRESHOLD:
|
| 85 |
+
return "Other", f"L1(0.00) -> L2(0.00)"
|
| 86 |
|
| 87 |
l2_tok, l2_mod = l2_manager.get_model(pred1)
|
| 88 |
|