Update app.py

app.py

@@ -158,11 +158,11 @@ def predict_fn(manual_dist, manual_stops, hour, day_name, route_id, start_str, e
         diff_stops = abs(map_stops - float(manual_stops))
         
         if diff_dist > 50 or diff_stops > 0: # 50m tolerance
-            status_tag = "ℹ️ Based on Manual Inputs (Modified)"
+            status_tag = "Manual Inputs"
         else:
-            status_tag = "✅ Based on Map Data"
+            status_tag = "Computed Data"
     else:
-        status_tag = "ℹ️ Based on Manual Inputs"
+        status_tag = "Manual Inputs"
 
     try:
         inputs = {
@@ -188,6 +188,7 @@ def predict_fn(manual_dist, manual_stops, hour, day_name, route_id, start_str, e
 with gr.Blocks(title="HK-TransitFlow-Net") as demo:
     gr.Markdown("# HK-TransitFlow-Net Demo")
     gr.Markdown("Predicts KMB/CTB bus travel time.")
+    gr.Markdown("Model: https://huggingface.co/WheelsTransit/HK-TransitFlow-Net")
     
     with gr.Row():
         with gr.Column():