changes

app.py

@@ -17,28 +17,23 @@ def analyze_emotion(image):
     if image is None:
         return "Upload an image", None
 
-    # Preprocess
-    if isinstance(image, np.ndarray):
-        image = Image.fromarray(image)
-    if image.mode != "RGB":
-        image = image.convert("RGB")
+    # Ensure RGB
+    image = image.convert("RGB")
 
     inputs = processor(images=image, return_tensors="pt")
     with torch.no_grad():
         outputs = model(**inputs)
         probs = F.softmax(outputs.logits, dim=-1)[0].numpy()
 
-    # Get top emotion
     top_idx = np.argmax(probs)
     top_emotion = EMOTIONS[top_idx]
 
-    # Prepare bar chart
     chart_data = {"emotion": EMOTIONS, "confidence": probs.tolist()}
-
     result_text = f"Predicted Emotion: {top_emotion} ({probs[top_idx]*100:.1f}%)"
 
     return result_text, chart_data
 
+
 # Build Gradio interface
 demo = gr.Interface(
     fn=analyze_emotion,
@@ -51,5 +46,6 @@ demo = gr.Interface(
     description="Upload a facial image and detect emotions (Angry, Disgust, Fear, Happy, Sad, Surprise, Neutral) using a Vision Transformer."
 )
 
+
 if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    demo.launch(server_name="0.0.0.0", server_port=7860, share=True)