Update app.py

app.py

@@ -3,12 +3,12 @@ from ultralytics import YOLO
 import cv2
 import numpy as np
 import os
+import requests
 
 # Ensure the model file is in the correct location
 model_path = "yolov8x-doclaynet-epoch64-imgsz640-initiallr1e-4-finallr1e-5.pt"
 if not os.path.exists(model_path):
     # Download the model file if it doesn't exist
-    import requests
     model_url = "https://huggingface.co/DILHTWD/documentlayoutsegmentation_YOLOv8_ondoclaynet/resolve/main/yolov8x-doclaynet-epoch64-imgsz640-initiallr1e-4-finallr1e-5.pt"
     response = requests.get(model_url)
     with open(model_path, "wb") as f:
@@ -30,9 +30,9 @@ def process_image(image):
 # Define the Gradio interface
 interface = gr.Interface(
     fn=process_image,
-    inputs=gr.inputs.Image(type="pil"),
-    outputs=[gr.outputs.Image(type="pil", label="Annotated Image"),
-             gr.outputs.Textbox(label="Detected Areas and Labels")]
+    inputs=gr.Image(type="pil"),
+    outputs=[gr.Image(type="pil", label="Annotated Image"),
+             gr.Textbox(label="Detected Areas and Labels")]
 )
 
 if __name__ == "__main__":