SAA+

app.py

@@ -6,6 +6,7 @@ os.system('pip install -e .')
 os.chdir('../SAM')
 os.system('pip install -e .')
 os.system('pip install opencv-python pycocotools matplotlib onnxruntime onnx ipykernel gradio loguru')
+os.system('pip install torch==1.10.0 torchvision==0.11.1 -f https://download.pytorch.org/whl/cu113/torch_stable.html')
 
 os.chdir('..')
 os.mkdir('weights')
@@ -23,10 +24,6 @@ import SAA as SegmentAnyAnomaly
 from utils.training_utils import *
 import os
 
-gpu_id = 0
-
-os.environ['CURL_CA_BUNDLE'] = ''
-os.environ['CUDA_VISIBLE_DEVICES'] = f"{gpu_id}"
 
 
 dino_config_file = 'GroundingDINO/groundingdino/config/GroundingDINO_SwinT_OGC.py'
@@ -35,7 +32,7 @@ sam_checkpoint = 'weights/sam_vit_h_4b8939.pth'
 box_threshold = 0.1
 text_threshold = 0.1
 eval_resolution = 1024
-device = f"cuda:0"
+device = f"cpu"
 root_dir = 'result'
 
 # get the model