Faisal commited on
Commit
488d84d
·
1 Parent(s): c1d8c8a

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("/kaggle/working/Brain-Tumor-Segmentation/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."""