update app.py

app.py

@@ -43,7 +43,7 @@ def get_figure(in_pil_img, in_results):
     for score, label, box in zip(in_results["scores"], in_results["labels"], in_results["boxes"]):
         selected_color = choice(COLORS)
 
-        x, y, w, h = torch.round(box[0]), torch.round(box[1]), torch.round(box[2]-box[0]), torch.round(box[3]-box[1])
+        x, y, w, h = torch.round(box[0]).item(), torch.round(box[1]).item(), torch.round(box[2]-box[0]).item(), torch.round(box[3]-box[1]).item()
         #x, y, w, h = int(box[0]), int(box[1]), int(box[2]-box[0]), int(box[3]-box[1])
 
         ax.add_patch(plt.Rectangle((x, y), w, h, fill=False, color=selected_color, linewidth=2))