remove zero gpu
Browse files
app.py
CHANGED
@@ -7,9 +7,9 @@ import gradio as gr
|
|
7 |
import cv2
|
8 |
import tempfile
|
9 |
from ultralytics import YOLO
|
10 |
-
import spaces
|
11 |
|
12 |
-
@spaces.GPU
|
13 |
def yolov12_inference(image, video, model_id, image_size, conf_threshold):
|
14 |
model = YOLO(model_id)
|
15 |
if image:
|
|
|
7 |
import cv2
|
8 |
import tempfile
|
9 |
from ultralytics import YOLO
|
10 |
+
# import spaces
|
11 |
|
12 |
+
# @spaces.GPU
|
13 |
def yolov12_inference(image, video, model_id, image_size, conf_threshold):
|
14 |
model = YOLO(model_id)
|
15 |
if image:
|