Update app.py

app.py

@@ -8,7 +8,7 @@ from transformers import AutoFeatureExtractor
 # Load the model from Hugging Face model hub
 model_id = "KabeerAmjad/food_classification_model"
 # Load ResNet50 model and adjust the final layer
-model = models.resnet50(pretrained=False)
+model = models.resnet50(pretrained=True)
 model.fc = nn.Linear(model.fc.in_features, 11)  # Adjust the output layer to match your number of classes
 
 # Load the weights from the Hugging Face model hub