omarelsayeed commited on
Commit
190bae6
1 Parent(s): 63b70c6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -2,6 +2,8 @@
2
  from ultralytics import ASSETS, YOLO, RTDETR
3
  import gradio as gr
4
  from huggingface_hub import snapshot_download
 
 
5
  model_dir = snapshot_download("omarelsayeed/DETR-ARABIC-DOCUMENT-LAYOUT-ANALYSIS") + "/Model.pt"
6
  model = RTDETR(model_dir)
7
 
 
2
  from ultralytics import ASSETS, YOLO, RTDETR
3
  import gradio as gr
4
  from huggingface_hub import snapshot_download
5
+ from PIL import Image
6
+
7
  model_dir = snapshot_download("omarelsayeed/DETR-ARABIC-DOCUMENT-LAYOUT-ANALYSIS") + "/Model.pt"
8
  model = RTDETR(model_dir)
9