Update app.py

app.py

@@ -15,15 +15,28 @@ def show_preds_image(image):
     
     image = cv2.imread(image_path)
     outputs = model.predict(source=image_path)
-    results = outputs[0].cpu().numpy()
-    for i, det in enumerate(results.boxes.xyxy):
+    #print("output>>>>>>>>>>>>>>>>>>>>>>>>", outputs)
+    results = outputs[0].boxes.xyxy.cpu().numpy()
+    for det in results:
+        x1, y1, x2, y2 = det[:4]
+        label = "Pothole"
         cv2.rectangle(
             image,
-            (int(det[0]), int(det[1])),
-            (int(det[2]), int(det[3])),
+            (int(x1), int(y1)),
+            (int(x2), int(y2)),
             color=(0, 0, 255),
             thickness=2,
-            lineType=cv2.LINE_AA
+            lineType=cv2.LINE_AA,
+        )
+        cv2.putText(
+            image,
+            label,
+            (int(x1), int(y1) - 10),
+            cv2.FONT_HERSHEY_SIMPLEX,
+            0.9,
+            (0, 0, 255),
+            2,
+            cv2.LINE_AA,
         )
     os.remove(image_path)  # Remove the temporary image file
     return cv2.cvtColor(image, cv2.COLOR_BGR2RGB)