Update app.py

app.py

@@ -5,11 +5,11 @@ import numpy as np
 from PIL import Image, ImageEnhance
 from ultralytics import YOLO
 
-# Load YOLOv11 Model
+
 model_path = "best.pt" 
 model = YOLO(model_path)
 
-# ---------------- Preprocessing Function ---------------- #
+
 def preprocessing(image):
     """Apply three enhancement filters, including brightness reduction, and resize."""
     image = Image.fromarray(np.array(image))
@@ -27,12 +27,12 @@ def preprocessing(image):
 
     return image
 
-# ---------------- Dummy Image Rotation Function ---------------- #
+
 def imageRotation(image):
     """Dummy function for image rotation."""
     return image
 
-# ---------------- Document Detection Function ---------------- #
+
 def detect_document(image):
     """Detects front and back of the document using YOLO."""
     image = np.array(image)
@@ -107,7 +107,7 @@ def predict(image):
         "back": back_result
     }
 
-    return detected_image, labels, api_results, cropped_images  # Return cropped images as output
+    return cropped_images, labels, api_results
 
 # ---------------- Gradio Interface ---------------- #
 iface = gr.Interface(