Update app.py

app.py

@@ -1,8 +1,8 @@
 import subprocess
 
-# subprocess.run(["pip", "install", "-e", "./models/GroundingDINO"])
-# subprocess.run(["pip", "install", "gradio==4.21.0"])
-# subprocess.run(["pip", "install", "fastapi==0.108.0"])
+subprocess.run(["pip", "install", "-e", "./models/GroundingDINO"])
+subprocess.run(["pip", "install", "gradio==4.21.0"])
+subprocess.run(["pip", "install", "fastapi==0.108.0"])
 
 
 
@@ -10,9 +10,9 @@ import gradio as gr
 
 from UniVAD.tools import process_image
 
-# subprocess.run(["wget", "-q","https://github.com/IDEA-Research/GroundingDINO/releases/download/v0.1.0-alpha/groundingdino_swint_ogc.pth"], check=True)
-# subprocess.run(["wget", "-q","https://huggingface.co/xinyu1205/recognize-anything-plus-model/resolve/main/ram_plus_swin_large_14m.pth"], check=True)
-# subprocess.run(["wget", "-q","https://huggingface.co/lkeab/hq-sam/resolve/main/sam_hq_vit_h.pth"], check=True)
+subprocess.run(["wget", "-q","https://github.com/IDEA-Research/GroundingDINO/releases/download/v0.1.0-alpha/groundingdino_swint_ogc.pth"], check=True)
+subprocess.run(["wget", "-q","https://huggingface.co/xinyu1205/recognize-anything-plus-model/resolve/main/ram_plus_swin_large_14m.pth"], check=True)
+subprocess.run(["wget", "-q","https://huggingface.co/lkeab/hq-sam/resolve/main/sam_hq_vit_h.pth"], check=True)
 
 import torch
 import torchvision.transforms as transforms
@@ -33,7 +33,7 @@ from UniVAD.models.grounded_sam import (
 
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-image_size = 224
+image_size = 335
 univad_model = UniVAD(image_size=image_size).to(device)