Update app.py

app.py

@@ -54,6 +54,7 @@ def predict_image(image):
     for result in results:
         boxes = result[:4]
         prob = result[4]
+        prob0 = round(prob, 2)
         classes = int(result[5])
         boxes = boxes/img_size               
 
@@ -66,7 +67,7 @@ def predict_image(image):
 
         if prob > 0.2:
             cv2.rectangle(temp_image, (x1, y1), (x2, y2), (0, 0, 255), 2)
-            label_text = f"{classes} {round(prob, 2)}"
+            label_text = f"{classes} {prob0}"
             cv2.putText(temp_image, label_text, (x1, y1), 0, 0.5, (0, 255, 0), 2)
 
             # Store prediction info in a JSON-compatible format