Update app.py

app.py

@@ -3,20 +3,35 @@ import sys
 import subprocess
 import torch
 
-# Clone only the runtime source repo if missing
-if not os.path.exists("clip-interrogator"):
-    subprocess.run(
-        [
-            "git",
-            "clone",
-            "-b",
-            "open-clip",
-            "https://github.com/pharmapsychotic/clip-interrogator.git",
-        ],
-        check=True,
-    )
 
-# Download cache files
+def clone_if_missing(path, repo, branch=None):
+    if os.path.exists(path):
+        return
+
+    cmd = ["git", "clone"]
+    if branch:
+        cmd += ["-b", branch]
+    cmd += [repo, path]
+
+    subprocess.run(cmd, check=True)
+
+
+clone_if_missing(
+    "src/blip",
+    "https://github.com/pharmapsychotic/BLIP.git",
+    branch="lib",
+)
+
+clone_if_missing(
+    "clip-interrogator",
+    "https://github.com/pharmapsychotic/clip-interrogator.git",
+    branch="open-clip",
+)
+
+sys.path.append("src/blip")
+sys.path.append("clip-interrogator")
+
+
 print("Download preprocessed cache files...")
 
 CACHE_URLS = [
@@ -32,8 +47,6 @@ os.makedirs("cache", exist_ok=True)
 for url in CACHE_URLS:
     subprocess.run(["wget", "-nc", url, "-P", "cache"], check=False)
 
-sys.path.append("src/blip")
-sys.path.append("clip-interrogator")
 
 import gradio as gr
 from clip_interrogator import Config, Interrogator