Update app.py

app.py

@@ -8,7 +8,7 @@ from typing import List, Optional, Tuple
 
 import gradio as gr
 import numpy as np
-from huggingface_hub import hf_hub_download, HfApi
+from huggingface_hub import hf_hub_download
 from unigaze.infer_runtime import UniGazeRuntime
 
 # --------------------------------------------------------------------------------------
@@ -25,7 +25,7 @@ CHECKPOINTS = [
 "unigaze_b16_joint.pth.tar",
 ]
 
-# 🔥 SAFE MAPPING (fixes your bug permanently)
+# 🔥 SAFE MAPPING (fixes mismatch issue)
 
 MODEL_MAP = {
 "unigaze_h14_joint.pth.tar": "unigaze/configs/model/mae_h_14_gaze.yaml",
@@ -39,7 +39,9 @@ Upload a short video or a single image.
 Checkpoint automatically loads correct model config (no mismatch errors).
 """
 
-sys.path.append(os.path.dirname(__file__))
+# Fix import path
+
+sys.path.append(os.path.dirname(**file**))
 
 # --------------------------------------------------------------------------------------
 
@@ -61,13 +63,14 @@ repo_type="model",
 )
 
 from functools import lru_cache
+
 @lru_cache(maxsize=3)
 def get_runtime(cfg_abs: str, ckpt_path: str):
 return UniGazeRuntime(cfg_abs, ckpt_path, device="cpu")
 
 # --------------------------------------------------------------------------------------
 
-# RUNNERS
+# Runners
 
 # --------------------------------------------------------------------------------------
 
@@ -88,7 +91,7 @@ cfg_abs = resolve_cfg_abs(cfg_path)
 rt = get_runtime(str(cfg_abs), ckpt_path)
 out = rt.predict_image(image_array)
 
-logs.append(f"[time] {time.time()-t0:.2f}s")
+logs.append(f"[time] {time.time() - t0:.2f}s")
 return out, "\n".join(logs)
 ```
 
@@ -125,16 +128,16 @@ gr.Markdown(f"# {TITLE}\n{DESC}")
 ```
 ckpt = gr.Dropdown(
     choices=CHECKPOINTS,
-    value=CHECKPOINTS[1],  # default L16 (safe)
-    label="Checkpoint"
+    value=CHECKPOINTS[1],  # default L16
+    label="Checkpoint",
 )
 
-# IMAGE
+# IMAGE TAB
 with gr.Tab("Image"):
-    inp = gr.Image(type="numpy")
+    inp = gr.Image(type="numpy", label="Input Image")
     btn = gr.Button("Run")
-    out = gr.Image()
-    logs = gr.Textbox(lines=10)
+    out = gr.Image(label="Output Image")
+    logs = gr.Textbox(label="Logs", lines=10)
 
     btn.click(
         fn=run_image,
@@ -142,13 +145,13 @@ with gr.Tab("Image"):
         outputs=[out, logs],
     )
 
-# VIDEO
+# VIDEO TAB
 with gr.Tab("Video"):
-    vin = gr.Video()
+    vin = gr.Video(label="Input Video")
     btn2 = gr.Button("Run")
-    vout = gr.Video()
-    img_out = gr.Image()
-    logs2 = gr.Textbox(lines=10)
+    img_out = gr.Image(label="Last Frame")
+    vout = gr.Video(label="Output Video")
+    logs2 = gr.Textbox(label="Logs", lines=10)
 
     btn2.click(
         fn=run_video,