Update app.py

app.py

@@ -7,9 +7,9 @@ import zipfile
 import pathlib
 import shutil
 
-# -------------------------------------------------------
-# 1) Load the AutoGluon predictor (pickled .pkl file)
-# -------------------------------------------------------
+# -----------------------------
+# 1) Load pickled AutoGluon predictor
+# -----------------------------
 pkl_path = hf_hub_download(
     repo_id="cassieli226/hw2-airline-automl",
     filename="autogluon_predictor.pkl",
@@ -19,9 +19,9 @@ pkl_path = hf_hub_download(
 with open(pkl_path, "rb") as f:
     predictor = cloudpickle.load(f)
 
-# -------------------------------------------------------
-# 2) load predictor from zipped directory for leaderboard
-# -------------------------------------------------------
+# -----------------------------
+# 2) load predictor directory for leaderboard
+# -----------------------------
 zip_path = hf_hub_download(
     repo_id="cassieli226/hw2-airline-automl",
     filename="autogluon_predictor_dir.zip",
@@ -35,12 +35,11 @@ if extract_dir.exists():
 with zipfile.ZipFile(zip_path, "r") as zf:
     zf.extractall(str(extract_dir))
 
-# Load predictor directory (if needed)
 predictor_dir = ag.TabularPredictor.load(str(extract_dir))
 
-# -------------------------------------------------------
-# 3) Define Gradio interface
-# -------------------------------------------------------
+# -----------------------------
+# 3) Gradio interface function
+# -----------------------------
 def predict_flight(stops, days_from_departure, flight_time, price, day_of_week, destination):
     X = pd.DataFrame({
         "Stops": [stops],
@@ -52,9 +51,12 @@ def predict_flight(stops, days_from_departure, flight_time, price, day_of_week,
     })
     return predictor.predict(X)[0]
 
+# -----------------------------
+# 4) Gradio UI
+# -----------------------------
 with gr.Blocks() as demo:
     gr.Markdown("# Flight Duration Predictor")
-    
+
     with gr.Row():
         with gr.Column():
             stops_in = gr.Slider("Stops", 0, 3, 1)
@@ -74,7 +76,7 @@ with gr.Blocks() as demo:
         outputs=[output]
     )
 
-# -------------------------------------------------------
-# 4) Launch app
-# -------------------------------------------------------
+# -----------------------------
+# 5) Launch
+# -----------------------------
 demo.launch()