Spaces:

Sa-m
/

YoloV5-Party-Symbol-Detector-V1

Sleeping

Sa-m commited on May 30, 2024

Commit

06949d1

verified ·

1 Parent(s): 71369b6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,7 +24,7 @@ def detect(inp):
  #im = im.resize((int(x * g) for x in im.size), Image.ANTIALIAS)  # resize
  results = model(inp,size=640)  # inference
  results.render()  # updates results.imgs with boxes and labels
- return Image.fromarray(results.imgs[0])
 inp = gr.Image(type="pil", label="Original Image")

  #im = im.resize((int(x * g) for x in im.size), Image.ANTIALIAS)  # resize
  results = model(inp,size=640)  # inference
  results.render()  # updates results.imgs with boxes and labels
+ return Image.fromarray(results.ims[0])
 inp = gr.Image(type="pil", label="Original Image")