Faisal commited on
Commit
30d9472
Β·
1 Parent(s): 90e84e7

Restore GPU version - remove CPU optimizations and restore GPU-compatible dependencies

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -39,7 +39,7 @@ temp_generation_config = GenerationConfig(
39
  # ----------------------------
40
  # YOLO MODEL LOADING
41
  # ----------------------------
42
- yolo_model = YOLO("MedSegVLM_with_DeepSeek/best.pt") # replace with your segmentation weights
43
 
44
  def inference(image_path: str):
45
  """Runs YOLO segmentation on an image and returns the annotated image."""
 
39
  # ----------------------------
40
  # YOLO MODEL LOADING
41
  # ----------------------------
42
+ yolo_model = YOLO("best.pt") # replace with your segmentation weights
43
 
44
  def inference(image_path: str):
45
  """Runs YOLO segmentation on an image and returns the annotated image."""