Update app.py

app.py

@@ -64,7 +64,7 @@ def detect_document(image):
 
     return Image.fromarray(image), labels, bounding_boxes
 
-# ---------------- Cropping Function ---------------- #
+
 def crop_image(image, bounding_boxes):
     """Crops detected bounding boxes from the image."""
     cropped_images = {}
@@ -76,7 +76,7 @@ def crop_image(image, bounding_boxes):
 
     return cropped_images
 
-# ---------------- Vision AI API Call (Dummy) ---------------- #
+
 def vision_ai_api(image, doc_type):
     """Dummy API call for Vision AI, returns a fake JSON response."""
     return {
@@ -85,7 +85,7 @@ def vision_ai_api(image, doc_type):
         "confidence": 0.99
     }
 
-# ---------------- Prediction Function ---------------- #
+
 def predict(image):
     """Pipeline: Preprocess -> Detect -> Crop -> Vision AI API."""
     processed_image = preprocessing(image)
@@ -109,7 +109,7 @@ def predict(image):
     single_image = cropped_images.get("front") or cropped_images.get("back") or detected_image   
     return single_image, labels, api_results
 
-# ---------------- Gradio Interface ---------------- #
+
 iface = gr.Interface(
     fn=predict, 
     inputs="image",