leosheck commited on
Commit
de062d4
·
verified ·
1 Parent(s): 2d728ff

Update octolyzer/segment/sloseg/slo_inference.py

Browse files
octolyzer/segment/sloseg/slo_inference.py CHANGED
@@ -107,7 +107,7 @@ def get_img_list_dataloader(img_list, batch_size=16, num_workers=0, pin_memory=F
107
 
108
  class SLOSegmenter:
109
 
110
- DEFAULT_MODEL_URL = 'https://huggingface.co/spaces/leosheck/SLO_vessel_analysis/resolve/main/weights/slosegmenter_weights.pth'
111
  DEFAULT_THRESHOLD = 0.5
112
  DEFAULT_MODEL_PATH = None
113
  # DEFAULT_MODEL_PATH = os.path.join(SCRIPT_PATH, r"weights/slosegmenter_weights.pth")
 
107
 
108
  class SLOSegmenter:
109
 
110
+ DEFAULT_MODEL_URL = 'https://huggingface.co/spaces/leosheck/SLO_vessel_analysis/resolve/main/octolyzer/segment/sloseg/weights/slosegmenter_weights.pth'
111
  DEFAULT_THRESHOLD = 0.5
112
  DEFAULT_MODEL_PATH = None
113
  # DEFAULT_MODEL_PATH = os.path.join(SCRIPT_PATH, r"weights/slosegmenter_weights.pth")