Update app.py

app.py

@@ -10,12 +10,12 @@ if not os.path.exists('model'):
     os.makedirs('model')
 
 # Load the pre-trained model
-model = models.resnet50(weights=None)
+model = models.seresnet50(weights=None)
 model.fc = torch.nn.Linear(2048, 7)  # Adjust for the number of emotion categories
 
 # Load the model
 try:
-    model.load_state_dict(torch.load('custom_resnet50_emotion_model.pth'))  # Adjust the path if needed
+    model.load_state_dict(torch.load('custom_resnet50_emotion_model_full.pth'))  # Adjust the path if needed
     print("Model loaded successfully.")
 except FileNotFoundError:
     print("Model file not found. Please check the path.")