Dricz commited on
Commit
09410a4
1 Parent(s): 89feddf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -28,11 +28,13 @@ def response2(image: gr.Image = None,image_size: gr.Slider = 640, conf_threshold
28
 
29
  render = render_result(model=model, image=image, result=results[0], rect_th = 1, text_th = 1)
30
 
31
- text = "Success"
32
 
33
- # for score, label, box in zip(result.scores, result.labels, result.boxes):
34
- # box = [round(i, 2) for i in box.tolist()]
35
- # text += f"Detected {model.config.id2label[label.item()]} with confidence {round(score.item(), 3)} at location {box}\n"
 
 
36
 
37
  return render,text
38
 
 
28
 
29
  render = render_result(model=model, image=image, result=results[0], rect_th = 1, text_th = 1)
30
 
31
+ text = ""
32
 
33
+ for r in results:
34
+ conf = r.boxes.conf
35
+ cls = r.boxes.cls
36
+
37
+ text += f"Detected {cls} with confidence {conf)}\n"
38
 
39
  return render,text
40