Update app.py

app.py

@@ -29,7 +29,7 @@ def process_img(image):
         result = model(source=image)
         if len(result[0].boxes)>0:
             vis = plots.plot_detection(image,boxes=result[0].boxes,
-                     class_names=class_names, min_score=0.2)
+                     class_names=result[0].names, min_score=0.2)
         else:
             vis = img
     return vis