Update app.py

app.py

@@ -2,12 +2,13 @@ import gradio as gr
 import torch
 from PIL import Image
 import json
+import os
 # import gdown
-
+yo=os.system('https://github.com/ultralytics/yolov5.git')
 
 # Model
 # model = torch.hub.load('ultralytics/yolov5', 'yolov5s')  # force_reload=True to update
-model = torch.hub.load("custom","best.pt") 
+model = torch.hub.load("yo","custom","best.pt") 
 
 def yolo(im, size=1024):
     g = (size / max(im.size))  # gain