omarelsayeed
commited on
Commit
•
190bae6
1
Parent(s):
63b70c6
Update app.py
Browse files
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 |
|