Update app.py

app.py

@@ -2,6 +2,7 @@
 import io
 import os
 import cv2
+import numpy as np
 import gradio as gr
 import matplotlib.pyplot as plt
 import requests, validators
@@ -102,7 +103,7 @@ def visualize_prediction(img, output_dict, threshold=0.5, id2label=None):
             )
 
     plt.axis("off")
-    return fig2img(plt.gcf())(img, output_dict, threshold=0.5, id2label=None):
+    return fig2img(plt.gcf())
     keep = output_dict["scores"] > threshold
     boxes = output_dict["boxes"][keep].tolist()
     scores = output_dict["scores"][keep].tolist()