Update app.py

app.py

@@ -4,15 +4,19 @@ from pixellib.torchbackend.instance import instanceSegmentation
 
 ins = instanceSegmentation()
 ins.load_model("https://github.com/ayoolaolafenwa/PixelLib/releases/download/0.2.0/pointrend_resnet50.pkl")
+target_classes = seg.select_target_classes(person=True)
+#seg.segmentImage("sample2.jpg", segment_target_classes= target_classes, show_bboxes=True,  output_image_name="a.jpg") 
+
+
 def seg(inp):
     
-    results, output = ins.segmentImage(f"{inp}", show_bboxes=True, output_image_name="output_image.jpg")
+    results, output = ins.segmentImage(f"{inp}", segment_target_classes= target_classes, show_bboxes=False, output_image_name="output_image.jpg")
     #results, output = ins.segmentImage("image.jpg", show_bboxes=True, output_image_name="result.jpg")
     print (results)
-    return "output_image.jpg"
+    return results['masks']
 
 gr.Interface(
     fn=seg,
     inputs=gr.Image(type='filepath'),
-    outputs=gr.Image()
+    outputs=gr.Gallery()
 ).launch()