Update app.py

app.py

@@ -42,8 +42,6 @@ def calculate_bounding_box_area(results):
     areas = []
     class_names = []
 
-    print(results)
-
     # Iterate through all bounding boxes
     for box in results[0].boxes:
         x1, y1, x2, y2 = map(int, box.xyxy[0].tolist())  # Bounding box coordinates
@@ -73,10 +71,7 @@ def process_image(uploaded_file):
 
 # Function to run YOLO inference
 def run_yolo(image):
-    image_cv = np.array(image)
-    image_cv = cv2.cvtColor(image_cv, cv2.COLOR_RGB2BGR)
-
-    results = model(image_cv)  # Run YOLOv8 detection
+    results = model(image)  # Run YOLOv8 detection with PIL image directly
 
     detected_classes, bounding_areas = calculate_bounding_box_area(results)