Update app.py

app.py

@@ -209,9 +209,9 @@ def get_output(image):
     with torch.no_grad():
         bbox,output_path = detect(opt)
     if os.path.exists(output_path):
-        return Image.open(output_path)
+        return np.array(Image.open(output_path))
     else:
-        return image
+        return np.array(image)
 
 gr.Interface(fn=get_output, 
          inputs=gr.Image(type = "pil", label="Your image"),