Update app.py

app.py

@@ -22,16 +22,13 @@ def detect_objects_on_image(image_path):
     results = model.predict(image_path)
     result = results[0]
     output = []
+
     for box in result.boxes:
-        x1, y1, x2, y2 = [
-            round(x) for x in box.xyxy[0].tolist()
-            
-        ]
+        x1, y1, x2, y2 = [round(x) for x in box.xyxy[0].tolist()]
         class_id = box.cls[0].item()
         prob = round(box.conf[0].item(), 2)
-        output.append([
-            x1, y1, x2, y2, result.names[class_id], prob
-        ])
+        output.append([x1, y1, x2, y2, result.names[class_id], prob])
+
         cv2.rectangle(
             image,
             (x1, y1),
@@ -40,10 +37,15 @@ def detect_objects_on_image(image_path):
             thickness=2,
             lineType=cv2.LINE_AA
         )
-        
-        cv2.putText(image,result.names[class_id],  (x1, y1), cv2.FONT_HERSHEY_SIMPLEX, 0.9, (36,255,12), 2)
+
+        # Change font and size
+        font = cv2.FONT_HERSHEY_SIMPLEX
+        font_size = 0.5  # Adjust the font size as needed
+        cv2.putText(image, result.names[class_id], (x1, y1), font, font_size, (36, 255, 12), 1)
+
     return cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
 
+
 inputs_image = [
     gr.components.Image(type="filepath", label="Input Image"),
 ]