updates

README.md

@@ -1,6 +1,6 @@
 ---
-title: Parasitic Egg Detection
-emoji: 🐢
+title: Parasite Egg Detection
+emoji: 🦠
 colorFrom: gray
 colorTo: green
 sdk: gradio

app.py

@@ -7,8 +7,7 @@ torch.hub.download_url_to_file('https://st4.depositphotos.com/3687893/27930/i/45
 torch.hub.download_url_to_file('https://sanangelo.tamu.edu/files/2021/06/Image_4_whipworm_egg.jpg', 'three.jpg')
 
 def para_func(image: gr.Image = None, image_size: gr.Slider = 640, conf_threshold: gr.Slider = 0.4, iou_threshold: gr.Slider = 0.50):
-    # model = YOLO('yolov8n.pt')  #yolov8 trained model 
-    model = YOLO('best.pt')  # custom trained model
+    model = YOLO('best.pt')  # Custom trained model
 
     # Perform object detection on the input image using YOLO model
     results = model.predict(image, conf=conf_threshold, iou=iou_threshold, imgsz=image_size)
@@ -17,7 +16,7 @@ def para_func(image: gr.Image = None, image_size: gr.Slider = 640, conf_threshol
     box = results[0].boxes
     print("Object type:", box.cls)
     print("Coordinates:", box.xyxy)
-    print("Probability:", box.conf)
+    print("Probability:", f'{box.conf * 100:.1f}%')
 
     # Render the output image with bounding boxes around detected objects
     render = render_result(model=model, image=image, result=results[0])