Hugging Face's logo

Spaces:
Dricz
/
Weapon_Detection_YOLOv8
Paused

App Files Community
Dricz commited on
Commit
5d64bc0
•
1 Parent(s): 6dedfce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ def image_preprocess(image):
30
  model = YOLO('best (1).pt')
31
 
32
  def response(image):
33
- res = Image.open(image)
34
  im_rgb = model(res)
35
  # im_rgb = Image.fromarray(im_rgb)
36
  return np.array(im_rgb)
 
30
  model = YOLO('best (1).pt')
31
 
32
  def response(image):
33
+ res = Image.fromarray(image)
34
  im_rgb = model(res)
35
  # im_rgb = Image.fromarray(im_rgb)
36
  return np.array(im_rgb)