Update app.py

app.py

@@ -23,9 +23,11 @@ label_mapping = {
 }
 
 # Función de predicción
-def predict(img_path):
-    img = PILImage.create(img_path)
-    
+def predict(img):
+    img = img.convert('L')  # Convertir la imagen a escala de grises
+    img = img.resize((48, 48))  # Redimensionar la imagen a (48, 48)
+    img = PILImage.create(img)  # Convertir a PILImage
+
     pred_emotion, pred_emotion_idx, probs_emotion = learn_emotion.predict(img)
     pred_sentiment, pred_sentiment_idx, probs_sentiment = learn_sentiment.predict(img)
     
@@ -53,7 +55,7 @@ examples = [
 
 iface = gr.Interface(
     fn=predict,
-    inputs=gr.Image(shape=(48, 48), image_mode='L'),
+    inputs=gr.Image(type='pil'),  # Eliminar el argumento 'shape'
     outputs=[gr.Label(label='Emoción'), gr.Label(label='Sentimiento')],
     title=title,
     examples=examples,
@@ -63,5 +65,3 @@ iface = gr.Interface(
 )
 
 iface.launch(enable_queue=True)
-
-