Update app.py

app.py

@@ -3,7 +3,7 @@ import tensorflow as tf
 import numpy as np
 from huggingface_hub import hf_hub_download
 
-# --- FIX: Download and Load Manually ---
+# 1. Download and Load Model
 print("Downloading model...")
 model_path = hf_hub_download(repo_id="WheelsTransit/HK-TransitFlow-Net", filename="hk_transit_flow_net.keras")
 
@@ -43,20 +43,32 @@ def predict_eta(distance_meters, num_stops, hour, day_name, route_id):
     except Exception as e:
         return f"Error: {str(e)}"
 
-# Build the Interface
-iface = gr.Interface(
-    fn=predict_eta,
-    inputs=[
-        gr.Number(label="Distance (meters)", value=5000),
-        gr.Number(label="Number of Stops", value=10),
-        gr.Slider(minimum=0, maximum=23, step=1, label="Hour of Day (0-23)", value=9),
-        gr.Dropdown(choices=list(DAY_MAP.keys()), label="Day of Week", value="Monday"),
-        gr.Textbox(label="Route ID", placeholder="968+1+...", value="UNKNOWN")
-    ],
-    outputs="text",
-    title="HK-TransitFlow-Net Demo",
-    description="Live inference for HK Bus ETA prediction.",
-    theme="soft"
-)
+# 2. Build the UI using Blocks (More stable than Interface)
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# HK-TransitFlow-Net Demo 🚌")
+    gr.Markdown("Live inference for HK Bus ETA prediction.")
+    
+    with gr.Row():
+        with gr.Column():
+            # Inputs
+            dist_input = gr.Number(label="Distance (meters)", value=5000)
+            stops_input = gr.Number(label="Number of Stops", value=10)
+            hour_input = gr.Slider(minimum=0, maximum=23, step=1, label="Hour of Day (0-23)", value=9)
+            day_input = gr.Dropdown(choices=list(DAY_MAP.keys()), label="Day of Week", value="Monday")
+            route_input = gr.Textbox(label="Route ID (Optional)", placeholder="968+1+...", value="UNKNOWN")
+            
+            predict_btn = gr.Button("Predict ETA", variant="primary")
+            
+        with gr.Column():
+            # Output
+            output_text = gr.Textbox(label="Estimated Travel Time", lines=1)
+
+    # Event Listener
+    predict_btn.click(
+        fn=predict_eta,
+        inputs=[dist_input, stops_input, hour_input, day_input, route_input],
+        outputs=output_text
+    )
 
-iface.launch()
+if __name__ == "__main__":
+    demo.launch()