Update app.py

app.py

@@ -3,26 +3,25 @@ import tensorflow as tf
 import cv2
 import numpy as np
 
-# Load model
+# Load the model
 model = tf.keras.models.load_model("FER_DATA.keras")
 
-# Define emotion labels
-emotions = ['Angry', 'Happy', 'Sad', 'Neutral']
-
+# Define the emotion prediction function
 def predict_emotion(image):
     image = cv2.cvtColor(image, cv2.COLOR_RGB2GRAY)
     image = cv2.resize(image, (48, 48))
     image = image / 255.0
     image = np.expand_dims(image, axis=-1)
     image = np.expand_dims(image, axis=0)
-    predictions = model.predict(image)
-    emotion = emotions[np.argmax(predictions)]
-    return emotion
+    prediction = model.predict(image)
+    emotion_index = np.argmax(prediction)
+    emotions = ['Angry', 'Happy', 'Sad', 'Neutral']
+    return emotions[emotion_index]
 
-# Gradio interface
+# Launch the Gradio interface
 iface = gr.Interface(
     fn=predict_emotion,
-    inputs=gr.Image(type="numpy", shape=(224, 224)),
+    inputs=gr.Image(type="numpy"),  # Removed shape
     outputs="text",
     title="MoodSync - Emotion Detection"
 )