Update app.py

app.py

@@ -1,10 +1,9 @@
-# !pip install ultralytics
 from ultralytics import ASSETS, YOLO, RTDETR
 import gradio as gr
 from huggingface_hub import snapshot_download
 from PIL import Image
 
-model_dir = snapshot_download("omarelsayeed/DETR-ARABIC-DOCUMENT-LAYOUT-ANALYSIS") + "/rtdetr_640.pt"
+model_dir = snapshot_download("omarelsayeed/DETR-ARABIC-DOCUMENT-LAYOUT-ANALYSIS") + "/rtdetr_1024.pt"
 model = RTDETR(model_dir)
 
 
@@ -16,7 +15,7 @@ def predict_image(img, conf_threshold, iou_threshold):
         iou=iou_threshold,
         show_labels=True,
         show_conf=True,
-        imgsz=640,
+        imgsz=1024,
     )
 
     for r in results: