Update app.py

app.py

@@ -41,10 +41,17 @@ def display_category(cls_name):
 
 # ── Load model ───────────────────────────────────────────────
 print("Loading model...")
-model_path = hf_hub_download(repo_id=MODEL_REPO, filename=MODEL_FILE)
+config_path  = hf_hub_download(repo_id=MODEL_REPO, filename="model_config.json")
+weights_path = hf_hub_download(repo_id=MODEL_REPO, filename="weights.weights.h5")
+
+with open(config_path) as f:
+    config = f.read()
+
+from tensorflow.keras.applications.efficientnet import preprocess_input
 import keras
 keras.config.enable_unsafe_deserialization()
-model = tf.keras.models.load_model(model_path, compile=False)
+model = tf.keras.models.model_from_json(config)
+model.load_weights(weights_path)
 print("Model ready!")
 
 # ── Feature extraction ───────────────────────────────────────