adding gradio

gradio_app.py

@@ -5,31 +5,31 @@ from pathlib import Path
 import gradio as gr
 import numpy as np
 
-# # -----------------------------
-# # Setup paths and env
-# # -----------------------------
-# HF_HOME = "/app/hf_cache"
-# os.environ["HF_HOME"] = HF_HOME
-# os.environ["TRANSFORMERS_CACHE"] = HF_HOME
-# os.makedirs(HF_HOME, exist_ok=True)
-
-# PRETRAINED_DIR = "/app/pretrained"
-# os.makedirs(PRETRAINED_DIR, exist_ok=True)
-
-# # -----------------------------
-# # Step 1: Optional Model Download
-# # -----------------------------
-# def download_models():
-#     expected_model = os.path.join(PRETRAINED_DIR, "RAFT/raft-things.pth")
-#     if not Path(expected_model).exists():
-#         print("⚙️ Downloading pretrained models...")
-#         try:
-#             subprocess.check_call(["bash", "download/download_models.sh"])
-#             print("✅ Models downloaded.")
-#         except subprocess.CalledProcessError as e:
-#             print(f"❌ Model download failed: {e}")
-#     else:
-#         print("✅ Pretrained models already exist.")
+# -----------------------------
+# Setup paths and env
+# -----------------------------
+HF_HOME = "/app/hf_cache"
+os.environ["HF_HOME"] = HF_HOME
+os.environ["TRANSFORMERS_CACHE"] = HF_HOME
+os.makedirs(HF_HOME, exist_ok=True)
+
+PRETRAINED_DIR = "/app/pretrained"
+os.makedirs(PRETRAINED_DIR, exist_ok=True)
+
+# -----------------------------
+# Step 1: Optional Model Download
+# -----------------------------
+def download_models():
+    expected_model = os.path.join(PRETRAINED_DIR, "RAFT/raft-things.pth")
+    if not Path(expected_model).exists():
+        print("⚙️ Downloading pretrained models...")
+        try:
+            subprocess.check_call(["bash", "download/download_models.sh"])
+            print("✅ Models downloaded.")
+        except subprocess.CalledProcessError as e:
+            print(f"❌ Model download failed: {e}")
+    else:
+        print("✅ Pretrained models already exist.")
 
 
 # -----------------------------