Update app.py

app.py

@@ -34,7 +34,7 @@ def response2(image: gr.Image = None,image_size: gr.Slider = 640, conf_threshold
     
     for r in results:
         conf = (r.boxes.conf)
-        cls = (r.boxes.cls)
+        cls = int(r.boxes.cls)
         xywh = (r.boxes.xywh)
         x = xywh[0]
         y = xywh[1]