Update app.py

app.py

@@ -4,6 +4,11 @@ from fastai.vision.all import *
 import matplotlib.pyplot as plt
 
 # Load the trained model
+def label_func(fname):
+    categories = ["Happy", "Sad", "Angry"]
+    category = fname.parts[-2]  # Extract the category name from the path
+    return category if category in categories else "unknown"
+
 learn = load_learner('emotion_model.pkl')
 
 # Define function to preprocess image