downloading model from the hub + new examples

app.py

@@ -1,9 +1,12 @@
 import gradio as gr
-import yolov5
+from huggingface_hub import hf_hub_download
 from PIL import Image
+import yolov5
 
 # Model
-model = yolov5.load('./yolo_datacat/best.pt')
+model_file = hf_hub_download(repo_id="HugoSchtr/yolov5_datacat", filename="best.pt")
+
+model = yolov5.load(model_file)
 
 def predict(im, threshold=0.50):
     # g = (size / max(im.size))  # gain