Sompote commited on
Commit
438f042
·
verified ·
1 Parent(s): 3089110

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ import cv2
8
  model = YOLO("best-3.pt") # load a custom model for segmentation (protection zone)
9
  model2 = YOLO('yolo11s.pt') # load a second model for object detection
10
 
11
- @spaces.GPU
12
  def process_image(image):
13
  # Gradio passes the image as RGB, so we'll work in RGB color space
14
  image_rgb = np.array(image)
 
8
  model = YOLO("best-3.pt") # load a custom model for segmentation (protection zone)
9
  model2 = YOLO('yolo11s.pt') # load a second model for object detection
10
 
11
+
12
  def process_image(image):
13
  # Gradio passes the image as RGB, so we'll work in RGB color space
14
  image_rgb = np.array(image)