update

app.py

@@ -6,6 +6,7 @@ import os
 import numpy as np
 from PIL import Image
 from dotenv import load_dotenv
+from huggingface_hub import hf_hub_download
 
 load_dotenv()
 
@@ -22,10 +23,15 @@ app.add_middleware(
 
 ANIMALS = ["Cat", "Dog", "Panda"]
 
-# It would've been better to use an environment variable to fix this line actually...
-model_path = os.getenv("MODEL_PATH")
-print(model_path)
-model = tf.keras.models.load_model(model_path)
+
+# Download your SavedModel from the Hub
+repo_id = "drgou/howest-deployathome"
+hf_hub_download(repo_id, filename="config.json", repo_type="model", local_dir="./model")
+hf_hub_download(repo_id, filename="metadata.json", repo_type="model", local_dir="./model")
+hf_hub_download(repo_id, filename="model.weights.h5", repo_type="model", local_dir="./model")
+
+# Load it
+model = tf.keras.models.load_model("./model")
 
 
 @app.post('/upload/image')

requirements.txt

@@ -6,4 +6,5 @@ tensorflow==2.19.0
 pillow==11.2.1
 uvicorn==0.34.2
 python-multipart==0.0.20
-python-dotenv==1.1.0
+python-dotenv==1.1.0
+huggingface_hub