Spaces:

akashAD
/

yolov5-classify

Runtime error

akashAD commited on Jan 24, 2023

Commit

7aa6ddc

•

1 Parent(s): 8a86358

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,9 +4,7 @@ import gradio as gr
 import requests
 from PIL import Image
-# weights='/content/drive/MyDrive/yolov5/yolov5s-cls.pt'
-# model = DetectMultiBackend(weights)
 model = torch.hub.load('ultralytics/yolov5', 'custom', 'yolov5m-cls.pt').eval()  # load from PyTorch Hub
 model.classify = True

 import requests
 from PIL import Image
+#load models from pytorch hub
 model = torch.hub.load('ultralytics/yolov5', 'custom', 'yolov5m-cls.pt').eval()  # load from PyTorch Hub
 model.classify = True