robot2no1 commited on
Commit
e51c291
·
verified ·
1 Parent(s): f6da317

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -11,8 +11,7 @@ model = None
11
  def load_model():
12
  global model
13
  if model is None:
14
- model_path = os.getenv('MODEL_PATH', 'yolo11x-seg.pt')
15
- model = YOLO(model_path)
16
  return model
17
 
18
  def segment_image(image, conf_threshold, iou_threshold, mask_threshold, line_thickness, use_retina_masks):
 
11
  def load_model():
12
  global model
13
  if model is None:
14
+ model = YOLO('yolo11x-seg.pt')
 
15
  return model
16
 
17
  def segment_image(image, conf_threshold, iou_threshold, mask_threshold, line_thickness, use_retina_masks):