Update app.py

app.py

@@ -5,9 +5,13 @@ import gradio as gr
 import requests
 from PIL import Image
 
-weights='yolov5s-cls.pt'
+# 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 
 
-model = DetectMultiBackend(weights)
 
 # load imagenet 1000 labels
 response = requests.get("https://git.io/JJkYN")
@@ -40,6 +44,6 @@ def predict(inp):
 
 gr.Interface(fn=predict, 
              inputs=gr.Image(type="pil"),
-             outputs="label",labels=labels).launch(debug=True)
+             outputs="label",labels=labels).launch()
             
              #outputs=gr.Label(num_top_classes=5))