Add model initialization with MD API key in app_v4.py

app_v4.py

@@ -20,6 +20,8 @@ DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 MAX_SEED = 1000000
 
 model_cache = {"models": None}
+md_api_key = os.getenv("MD_KEY")
+model = md.vl(api_key=md_api_key)
 
 @spaces.GPU(duration=12)  # This function gets priority for GPU access
 def load_warm_models():
@@ -42,10 +44,9 @@ def get_model():
 # # subprocess.run("rm -rf /data-nvme/zerogpu-offload/*", env={}, shell=True)
 @spaces.GPU()
 def init_space():
-    global pipe, model
+    global pipe
     pipe = get_model()["pipeline"]
-    model = get_model()["captioner"]
-    return pipe, model
+    return pipe
 
 @spaces.GPU(duration=12)
 @torch.no_grad()