Update app.py

app.py

@@ -29,15 +29,15 @@ def response2(image: gr.Image = None,image_size: gr.Slider = 640, conf_threshold
 
     box = results[0].boxes
 
-    # render = render_result(model=model, image=image, result=results[0], rect_th = 1, text_th = 1
-    for i, r in enumerate(results):
+    render = render_result(model=model, image=image, result=results[0], rect_th = 1, text_th = 1
+    # for i, r in enumerate(results):
       
-    # Plot results image
-      im_bgr = r.plot()  
-      im_rgb = im_bgr[..., ::-1]  # Convert BGR to RGB
+    # # Plot results image
+    #   im_bgr = r.plot()  
+    #   im_rgb = im_bgr[..., ::-1]  # Convert BGR to RGB
     
-    # im_rgb = Image.fromarray(im_rgb)
-    return im_rgb
+    # # im_rgb = Image.fromarray(im_rgb)
+    return render
 
 
 inputs = [