kevinconka commited on
Commit
efb936a
·
verified ·
1 Parent(s): 69f95c1

disabled Zero GPU for Gradio 5 beta

Browse files
Files changed (1) hide show
  1. model_yolov5.py +2 -2
model_yolov5.py CHANGED
@@ -1,4 +1,4 @@
1
- import spaces
2
  import numpy as np
3
  import yolov5
4
  from yolov5.utils.plots import Annotator, colors
@@ -12,7 +12,7 @@ def load_model(model_path, img_size=640):
12
  return model
13
 
14
 
15
- @spaces.GPU
16
  def inference(model, image):
17
  """Run inference on image and return annotated image."""
18
  results = model(image, size=model.img_size)
 
1
+ # import spaces
2
  import numpy as np
3
  import yolov5
4
  from yolov5.utils.plots import Annotator, colors
 
12
  return model
13
 
14
 
15
+ # @spaces.GPU
16
  def inference(model, image):
17
  """Run inference on image and return annotated image."""
18
  results = model(image, size=model.img_size)