Update app.py

app.py

@@ -20,7 +20,7 @@ def predict_language(text, top_k=5):
     probs = torch.nn.functional.softmax(logits, dim=-1).squeeze()
     top_probs, top_indices = torch.topk(probs, top_k)
     
-    results = [f"{id2label[str(idx.item())]}: {prob:.4f}" for prob, idx in zip(top_probs, top_indices)]
+    results = [f"{id2label[idx.item()]}: {prob:.4f}" for prob, idx in zip(top_probs, top_indices)]
     return "\n".join(results)
 
 # Create Gradio interface