add code to download the model downloading from hf

app.py

@@ -1,9 +1,13 @@
 import gradio as gr
 import PIL.Image as Image
 from ultralytics import YOLO
+from huggingface_hub import hf_hub_download
 
 
-model = YOLO('./models/best.pt')
+REPO_ID = 'vaivTA/yolov8x_doclaynet'
+FILENAME = "weights/best.pt"
+hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
+model = YOLO('./weights/best.pt')
 
 def infer(img_path):
     results = model(img_path)

models/best.pt

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:1dd302218b154d57bb4971bb9c5908d64484af51703a90bd427377841a481811
-size 6258543