Update app.py

app.py

@@ -59,17 +59,22 @@ def emo_preprocess(image):
 
 # Inference function
 def predict_emotion(image):
-    image = Image.open(image).convert("RGB")
-    image = emo_preprocess(image)
+    # If the image is passed as a PIL Image, you can directly use it
+    if isinstance(image, Image.Image):  
+        img = image.convert("RGB")
+    else:
+        img = Image.open(image).convert("RGB")  # In case the input is a path or something else
+    
+    img = emo_preprocess(img)
 
     with torch.no_grad():
-        outputs = model(image)
+        outputs = model(img)
         predicted = outputs.argmax(1).item()
 
     emotion = idx2label[predicted]
     emoji = emotion_emoji.get(emotion, "❓")  # Default to "?" if no emoji found
     return f"{emotion} {emoji}"
-
+    
 # Create Gradio interface
 iface = gr.Interface(
     fn=predict_emotion,