Dricz commited on
Commit
89feddf
1 Parent(s): 86d95b2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -28,11 +28,11 @@ 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 = ""
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 = "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