Update app.py

app.py

@@ -83,7 +83,7 @@ def iou(boxes1, boxes2):
 def predict(img, conf_thresh=0.4):
     img_input = [img_transforms(img)]
     _, pred = model(img_input)
-    pred = inter_class_nms(pred['boxes'], pred['scores'])
+    pred = [inter_class_nms(pred[0]['boxes'], pred[0]['scores'])]
     out_img = img.copy()
     draw = ImageDraw.Draw(out_img)
     font = ImageFont.truetype("res/Tuffy_Bold.ttf", 25)