Upload 7 files

main.py

@@ -44,7 +44,7 @@ def crop_image(model, original_image):
 
     Inputs:
         model (YOLO): The YOLO model used for object detection.
-        image_file (str): Path to the image file to be processed.
+        original_image (PIL.image): The image to be processed.
 
     Returns:
         PIL.Image: The cropped image containing the detected table.
@@ -70,10 +70,11 @@ def process_image(model, image):
     Process the uploaded image with YOLO model and draw bounding boxes with class-specific colors.
     
     Inputs:
-    uploaded_image (UploadedFile): The image file uploaded through Streamlit.
-    
+        model: The trained YOLO model
+        image: The image file uploaded through Streamlit.
+
     Returns:
-    PIL.Image: The processed image with bounding boxes and labels.
+        PIL.Image: The processed image with bounding boxes and labels.
     """
     colors = {'title': (255, 0, 0),
             'text': (0, 255, 0),