SandiyoS commited on
Commit
d3dafca
·
verified ·
1 Parent(s): 7c8170e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -38,12 +38,12 @@ logger = logging.getLogger(__name__)
38
  # ===================== CONFIG =====================
39
  class Config:
40
  device = "cuda" if torch.cuda.is_available() else "cpu"
41
- yolo_model_path = "yolo/best.pt"
42
  yolo_conf = 0.25
43
  yolo_img_size = 640
44
  yolo_min_crop_size = 32
45
  yolo_min_conf = 0.3
46
- lora_model_dir = "icon_model_hybrid"
47
  model_name = "ViT-L-16-SigLIP-256"
48
 
49
  CONFIG = Config()
 
38
  # ===================== CONFIG =====================
39
  class Config:
40
  device = "cuda" if torch.cuda.is_available() else "cpu"
41
+ yolo_model_path = "./yolo/best.pt"
42
  yolo_conf = 0.25
43
  yolo_img_size = 640
44
  yolo_min_crop_size = 32
45
  yolo_min_conf = 0.3
46
+ lora_model_dir = "./icon_model_hybrid"
47
  model_name = "ViT-L-16-SigLIP-256"
48
 
49
  CONFIG = Config()