Updated handler to download model from Hugging Face

handler.py

@@ -1,16 +1,32 @@
 import os
-import subprocess
 import torch
+import requests
 import torchvision.transforms as transforms
 from PIL import Image
 import json
 
-# Fetch model from Git LFS if not downloaded
-MODEL_PATH = "model_scripted_efficientnet.pt"
+# Hugging Face model URL
+HF_REPO = "your-username/your-model-name"
+MODEL_FILENAME = "model_scripted_efficientnet.pt"
+MODEL_PATH = f"/repository/{MODEL_FILENAME}"
 
-if not os.path.exists(MODEL_PATH):
-    print("Model file not found! Fetching from Git LFS...")
-    subprocess.run(["git", "lfs", "pull"], check=True)
+def download_model():
+    """Download model file from Hugging Face if it does not exist."""
+    if not os.path.exists(MODEL_PATH):
+        print("Model file not found! Downloading from Hugging Face...")
+        model_url = f"https://huggingface.co/{HF_REPO}/resolve/main/{MODEL_FILENAME}"
+        response = requests.get(model_url, stream=True)
+
+        if response.status_code == 200:
+            with open(MODEL_PATH, "wb") as f:
+                for chunk in response.iter_content(chunk_size=8192):
+                    f.write(chunk)
+            print("Model downloaded successfully!")
+        else:
+            raise RuntimeError(f"Failed to download model, status code: {response.status_code}")
+
+# Ensure model is downloaded before loading
+download_model()
 
 # Load Model
 device = "cuda" if torch.cuda.is_available() else "cpu"