Hugging Face's logo

Spaces:
akashAD
/
yolov5-classify
Runtime error

App Files Community
akashAD commited on
Commit
7aa6ddc
·
1 Parent(s): 8a86358

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -4,9 +4,7 @@ import gradio as gr
4
  import requests
5
  from PIL import Image
6
 
7
- # weights='/content/drive/MyDrive/yolov5/yolov5s-cls.pt'
8
-
9
- # model = DetectMultiBackend(weights)
10
 
11
  model = torch.hub.load('ultralytics/yolov5', 'custom', 'yolov5m-cls.pt').eval() # load from PyTorch Hub
12
  model.classify = True
 
4
  import requests
5
  from PIL import Image
6
 
7
+ #load models from pytorch hub
 
 
8
 
9
  model = torch.hub.load('ultralytics/yolov5', 'custom', 'yolov5m-cls.pt').eval() # load from PyTorch Hub
10
  model.classify = True