Update app.py

app.py

@@ -18,7 +18,7 @@ def classify_emotion(image: Image.Image) -> str:
         outputs = model(**inputs)
         pred = torch.argmax(outputs.logits, dim=1).item()
 
-    label = model.config.id2label[str(pred)]
+    label = model.config.id2label[pred]
     return f"🧠 Predicted Emotion: **{label}**"
 
 # Gradio interface