Update app.py

app.py

@@ -1,16 +1,29 @@
+import os
 import torch
 import torchvision.transforms as transforms
 import torchvision.models as models
 from PIL import Image
 import json
 import gradio as gr
+import requests
+
+# Path to the model file and Hugging Face URL
+model_path = 'food_classification_model.pth'
+model_url = "https://huggingface.co/KabeerAmjad/food_classification_model/resolve/main/food_classification_model.pth"
+
+# Download the model file if it's not already available
+if not os.path.exists(model_path):
+    print(f"Downloading the model from {model_url}...")
+    response = requests.get(model_url)
+    with open(model_path, 'wb') as f:
+        f.write(response.content)
+    print("Model downloaded successfully.")
 
 # Load the model with updated weights parameter
 model = models.resnet50(weights=models.ResNet50_Weights.DEFAULT)
 model.eval()  # Set model to evaluation mode
 
 # Load the model's custom state_dict
-model_path = 'food_classification_model.pth'
 try:
     state_dict = torch.load(model_path, map_location=torch.device('cpu'))
     model.load_state_dict(state_dict)