Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -44,7 +44,10 @@ def classify_text(text, model_choice):
|
|
| 44 |
outputs = model(input_ids, attention_mask=attention_mask).logits
|
| 45 |
if "multiclass" in model_choice:
|
| 46 |
probabilities = torch.softmax(outputs, dim=1).cpu().numpy()[0]
|
| 47 |
-
|
|
|
|
|
|
|
|
|
|
| 48 |
predicted_index = probabilities.argmax()
|
| 49 |
predicted_label = labels[predicted_index]
|
| 50 |
confidence = probabilities[predicted_index] * 100
|
|
|
|
| 44 |
outputs = model(input_ids, attention_mask=attention_mask).logits
|
| 45 |
if "multiclass" in model_choice:
|
| 46 |
probabilities = torch.softmax(outputs, dim=1).cpu().numpy()[0]
|
| 47 |
+
if model_choice == "ruSpamNS_v18_multiclass":
|
| 48 |
+
labels = ["袧袝 小袩袗袦", "小袩袗袦", "袧袝袛袙袠袞袠袦袨小孝鞋/孝袨袙袗袪蝎", "袙袗袣袗袧小袠袠", "袪袝袣袥袗袦袗/校小袥校袚袠"]
|
| 49 |
+
else:
|
| 50 |
+
labels = ["袧袝 小袩袗袦", "小袩袗袦", "袧袝袛袙袠袞袠袦袨小孝鞋", "袙袗袣袗袧小袠袠", "袩袪袨袛袗袞袗"]
|
| 51 |
predicted_index = probabilities.argmax()
|
| 52 |
predicted_label = labels[predicted_index]
|
| 53 |
confidence = probabilities[predicted_index] * 100
|