Update app.py

app.py

@@ -1,12 +1,14 @@
 import gradio as gr
 import tensorflow as tf
 import numpy as np
-from huggingface_hub import from_pretrained_keras
+from huggingface_hub import hf_hub_download
 
-# 1. Download the Model from your Repository
-# This connects this Space to your uploaded model
+# --- FIX: Download and Load Manually ---
 print("Downloading model...")
-model = from_pretrained_keras("WheelsTransit/HK-TransitFlow-Net")
+model_path = hf_hub_download(repo_id="WheelsTransit/HK-TransitFlow-Net", filename="hk_transit_flow_net.keras")
+
+print("Loading Keras model...")
+model = tf.keras.models.load_model(model_path)
 
 # Helper to map day names to integers
 DAY_MAP = {