Update app.py

app.py

@@ -75,7 +75,7 @@ def predict(text):
     pred_task1 = torch.argmax(logits_task1, dim=1).item()
     pred_task2 = torch.argmax(logits_task2, dim=1).item()
     
-    return "Тема": get_key_by_value(label_mapping_type, pred_task1)
+    return {"Тема": get_key_by_value(label_mapping_type, pred_task1)}
 
 iface = gr.Interface(
     fn=predict,