Display confidence

app.py

@@ -68,22 +68,22 @@ def predict_image(image, threshold):
 
         points = [(coord[0], coord[1]) for coord in coords]
         draw.polygon(points, outline="red", width=LINE_WIDTH)
-        draw.text((points[0][0], points[0][1]), detection['class_name'], fill="red")
+        draw.text((points[0][0], points[0][1]), detection['class_name'] + ' | ' + str(round(detection['confidence'], 3)), fill="red")
 
     return image, img.shape, len(detections), duration
 
 
 # Define example images and their true labels for users to choose from
 example_data = [
-    ["./demo/12ab97857.jpg", 0.8],
-    ["./demo/82f13510a.jpg", 0.8],
-    ["./demo/836f35381.jpg", 0.8],
-    ["./demo/848d2afef.jpg", 0.8],
-    ["./demo/911b25478.jpg", 0.8],
-    ["./demo/b86e4046f.jpg", 0.8],
-    ["./demo/ce2220f49.jpg", 0.8],
-    ["./demo/d9762ef5e.jpg", 0.8],
-    ["./demo/fa613751e.jpg", 0.8],
+    ["./demo/12ab97857.jpg", 0.6],
+    ["./demo/82f13510a.jpg", 0.6],
+    ["./demo/836f35381.jpg", 0.6],
+    ["./demo/848d2afef.jpg", 0.6],
+    ["./demo/911b25478.jpg", 0.6],
+    ["./demo/b86e4046f.jpg", 0.6],
+    ["./demo/ce2220f49.jpg", 0.6],
+    ["./demo/d9762ef5e.jpg", 0.6],
+    ["./demo/fa613751e.jpg", 0.6],
     # Add more example images and thresholds as needed
 
 ]
@@ -96,7 +96,7 @@ css = """
   } 
 """
 
-TITLE = "Ship detection on Optical Satellite images (Oriented Bounding Boxes)"
+TITLE = "Ship detection on SPOT Satellite images (Oriented Bounding Boxes)"
 
 # Define the Gradio Interface
 demo = gr.Blocks(title=TITLE, css=css).queue()