Update app.py

app.py

@@ -8,6 +8,7 @@ from PIL import Image
 
 model = torch.hub.load('ultralytics/yolov5', 'custom', 'yolov5m-cls.pt').eval()  # load from PyTorch Hub
 model.classify = True 
+model.conf = 0.40
 
 
 # load imagenet 1000 labels
@@ -41,6 +42,6 @@ def predict(inp):
 
 gr.Interface(fn=predict, 
              inputs=gr.Image(type="pil"),
-             outputs="label",labels=labels).launch()
+             outputs=gr.Label(num_top_classes=7)).launch()
             
-             #outputs=gr.Label(num_top_classes=5))
+