Update app.py

app.py

@@ -9,8 +9,10 @@ model = YOLO(YOLO_MODEL_PATH, task='detect').to("cpu")  # Force CPU usage
 def resize_and_pad(image, target_size=(640, 640)):
     """Resize image while keeping aspect ratio and padding to fit target size."""
     original_size = image.size  # (width, height)
-    image.thumbnail((target_size[0], target_size[1]), Image.ANTIALIAS)  # Resize while keeping aspect ratio
-
+    
+    # Use LANCZOS instead of ANTIALIAS (Fix for Pillow v10+)
+    image.thumbnail(target_size, Image.LANCZOS)  
+    
     # Create a new white background image
     new_image = Image.new("RGB", target_size, (255, 255, 255))
     
@@ -54,17 +56,21 @@ def detect_lines(image):
 
     return image_with_boxes
 
-# Define Gradio interface
+# Define Gradio interface with two-column layout
 with gr.Blocks() as iface:
-    gr.Markdown("# Text Line Detection")
-    gr.Markdown("## Upload an image to detect text lines")
+    gr.Markdown("# Text Line Detection with YOLOv8")
+    gr.Markdown("Upload an image and detect lines of text.")
 
-    with gr.Tab("Upload Image"):
-        gr.Markdown("Upload an image, and the YOLO model will detect lines of text.")
-        image_input = gr.Image(type="numpy", label="Upload an image")
-        image_output = gr.Image(type="pil", label="Detected lines")
-        
-        image_input.upload(detect_lines, inputs=image_input, outputs=image_output)
+    with gr.Row():
+        with gr.Column(scale=1):
+            gr.Markdown("### Upload Image")
+            image_input = gr.Image(type="numpy", label="Upload an image")
+
+        with gr.Column(scale=1):
+            gr.Markdown("### Detected Text Lines")
+            image_output = gr.Image(type="pil", label="Detected lines")
+    
+    image_input.upload(detect_lines, inputs=image_input, outputs=image_output)
 
 # Launch Gradio interface
 iface.launch()