Update app.py

app.py

@@ -180,10 +180,10 @@ def process_image(image):
     bounding_img = Image.open(bounding_path)
     return bounding_img, before_text, text
 
-# Define Gradio Interface
+
 interface = gr.Interface(
-    fn=process_image,  # Call the process_image function
-    inputs=gr.Image(type="pil"),  # Expect an image input
+    fn=process_image,
+    inputs=gr.Image(type="pil"),
     outputs=[
         gr.Image(type="pil", label="Bounding Box Image"),
         gr.Textbox(label="Extracted Text (Custom trained YOLO Object Detection + TrOCR Vision Transformer)"),
@@ -193,6 +193,5 @@ interface = gr.Interface(
     description="Upload an image to detect text regions with YOLO, merge bounding boxes, and extract text using TrOCR which is then preprocessed with Roberta for contextual understanding.",
 )
 
-# Launch the interface
 if __name__ == "__main__":
     interface.launch(share=True)