bug fixed cv2

app.py

@@ -14,10 +14,11 @@ from detectron2.utils.visualizer import Visualizer
 from detectron2.data import MetadataCatalog
 
 
-def predict(image):
+def predict(pil_image):
     print(datetime.datetime.now())
-    # Convert PIL Image to numpy array
-    input_image = cv2.imread(image.name)
+    # Convert PIL Image to CV2
+    open_cv_image = np.array(pil_image)
+    input_image = open_cv_image[:, :, ::-1].copy()
     # Initialise model
     cfg = get_cfg()
     cfg.merge_from_file(model_zoo.get_config_file("COCO-PanopticSegmentation/panoptic_fpn_R_101_3x.yaml"))
@@ -42,6 +43,6 @@ image_input = gr.Image(type="pil", label="Input Image")
 
 iface = gr.Interface(fn=predict,
                      inputs=[image_input],
-                     outputs=gr.Image(),
+                     outputs=gr.Image(type="pil"),
                      examples=["examples/6178_3003_0.jpg"])
 iface.launch()