changed main.py

main.py

@@ -7,7 +7,6 @@ from fastapi.middleware.cors import CORSMiddleware
 from fastapi import FastAPI
 from huggingface_hub import hf_hub_download
 
-# test
 app = FastAPI()
 
 app.add_middleware(
@@ -21,18 +20,14 @@ app.add_middleware(
 # Animal names here
 ANIMALS = ['Cat', 'Dog', 'Panda']
 
-model_path = hf_hub_download(
-    repo_id="RubenVierstraete/masterclass-2025",
-    filename="model.keras",
-    repo_type="model",
-    cache_dir="/tmp/hf_cache"
-)
-
-model = tf.keras.models.load_model(model_path)
-
-
-
+# 1) download your SavedModel from the Hub
+repo_id = "RubenVierstraete/masterclass-2025"
+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")
 
+# 2) load it
+model = tf.keras.models.load_model("./model")
 
 
 @app.post('/upload/image')