update

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-# from models.vsa_model import VisionSearchAssistant
+from models.vsa_model import VisionSearchAssistant
 from models.vsa_prompt import COCO_CLASSES
 
 
@@ -99,10 +99,10 @@ with gr.Blocks() as app:
     )
 
 
-# vsa = VisionSearchAssistant(
-#     ground_device = "cuda:0",
-#     vlm_device="cuda:0",
-#     vlm_load_4bit=True,
-# )
+vsa = VisionSearchAssistant(
+    ground_device = "cuda:0",
+    vlm_device="cuda:0",
+    vlm_load_4bit=True,
+)
 # Launch the app
 app.launch()