Update app.py

app.py

@@ -16,11 +16,9 @@ def detect(img):
     if isinstance(img,str):
         img = get_url_img(img) if img.startswith('http') else Image.open(img).convert('RGB')
     result = model.predict(source=img)
-    if len(result[0].boxes.data)>0:
-        print(result)
-        vis = img
-        # vis = plots.plot_detection(img,boxes=result[0].boxes.data,
-        #              class_names=class_names, min_score=0.2)
+    if len(result[0].boxes)>0:
+        vis = plots.plot_detection(img,boxes=result[0].boxes,
+                     class_names=class_names, min_score=0.2)
     else:
         vis = img
     return vis