Update app.py

app.py

@@ -11,10 +11,15 @@ target_classes = ins.select_target_classes(person=True)
 
 def seg(inp):
     out_box=[]
-    results, output = ins.segmentImage(f"{inp}", segment_target_classes= target_classes, show_bboxes=False, output_image_name="output_image.jpg")
+    results, output = ins.segmentImage(f"{inp}",
+                                       segment_target_classes= target_classes,
+                                       show_bboxes=False,
+                                       extract_segmented_objects= True,
+                                       save_extracted_objects=True,
+                                       output_image_name="output_image.jpg")
     #results, output = ins.segmentImage("image.jpg", show_bboxes=True, output_image_name="result.jpg")
     print (results)
-    for image in results['masks']:
+    for image in results['extracted_objects']:
         out_box.append(Image.fromarray(image))
     return out_box