Spaces:
Runtime error
Runtime error
| import gradio as gr | |
| import os | |
| import torch | |
| #Repo.clone_from("https://github.com/ultralytics/yolov5", "./content") | |
| #subprocess.run(["git","clone", "https://github.com/ultralytics/yolov5"]) | |
| '''gdd.download_file_from_google_drive(file_id='1xJcr1zuuoC6bbu6wGvmyw5dl0mUEi-Jn', | |
| dest_path='content/train_data.zip', | |
| unzip=True) | |
| gdd.download_file_from_google_drive(file_id='1UJ9Y2mw2KBRY0tO-TnBs5RALLya6C2dW', | |
| dest_path='content/video.zip', | |
| unzip=True) ''' | |
| #subprocess.run(["mv","content/custom_data.yaml","./yolov5/data"]) | |
| def find(name, path): | |
| for root, dirs, files in os.walk(path): | |
| if name in files: | |
| return os.path.join(root, name) | |
| model = torch.hub.load('ultralytics/yolov5', 'custom', path='/content/best.pt', force_reload=True) | |
| def detect(inp): | |
| output=model(inp) | |
| #outPath= | |
| #out_path='./yolov5/runs/detect/exp/'+inp_vid.split('/')[2] | |
| return output | |
| inp=gr.inputs.Image() | |
| output=gr.outputs.Image() | |
| io=gr.Interface(fn=detect, inputs=inp, outputs=output, title='Party Symbol Detection',examples=['content/4.jpg']) | |
| io.launch(debug=True,share=True) | |