sudip2003 commited on
Commit
a7a7037
·
verified ·
1 Parent(s): 7ebdd37

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -95,6 +95,7 @@ from ultralytics import YOLO
95
  face_model = YOLO(r"face_det.pt")
96
  seg_model = YOLO(r"seg_model.pt")
97
 
 
98
  def process_image(image):
99
  image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
100
  face_results = face_model.predict(image, conf=0.5, save=True)
 
95
  face_model = YOLO(r"face_det.pt")
96
  seg_model = YOLO(r"seg_model.pt")
97
 
98
+
99
  def process_image(image):
100
  image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
101
  face_results = face_model.predict(image, conf=0.5, save=True)