wzzanthony7 commited on
Commit
7168f17
·
verified ·
1 Parent(s): 3d0fba7
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -28,8 +28,12 @@ def RoboFlowGetOutlineBoxesPIL(pil_img):
28
  return result
29
 
30
  def localOutlineBox(image):
31
- model_name = "vt_dataset_yolov12_v7_weights.pt"
32
- model = YOLO(model_path)
 
 
 
 
33
  yolo_ret = model(image, verbose=False)
34
  useful_ret = yolo_ret[0]
35
  names = model.names
 
28
  return result
29
 
30
  def localOutlineBox(image):
31
+ model_name = "./vt_dataset_yolov12_v7_weights.pt"
32
+ if os.path.exists(model_name):
33
+ print("model exists")
34
+ else:
35
+ print("model is not available")
36
+ model = YOLO(model_name)
37
  yolo_ret = model(image, verbose=False)
38
  useful_ret = yolo_ret[0]
39
  names = model.names