Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import pixellib
 from pixellib.torchbackend.instance import instanceSegmentation
 from PIL import Image
+import cv2 
 
 ins = instanceSegmentation()
 ins.load_model("https://github.com/ayoolaolafenwa/PixelLib/releases/download/0.2.0/pointrend_resnet50.pkl")
@@ -20,7 +21,7 @@ def seg(inp):
     #results, output = ins.segmentImage("image.jpg", show_bboxes=True, output_image_name="result.jpg")
     print (results)
     for image in results['extracted_objects']:
-        out_box.append(Image.fromarray(image))
+        out_box.append(cv2.cvtColor(image, cv2.COLOR_BGR2RGB))
     return out_box
 
 gr.Interface(