Upload MyPipe.py

MyPipe.py

@@ -43,7 +43,7 @@ class RMBGPipe(Pipeline):
     def postprocess(self, inputs, out_name=""):
         result = inputs.pop("result")
         orig_im_size = inputs.pop("orig_im_size")
-        orig_image = inputs.pop("orig_image")
+        orig_image = inputs.pop("orig_im")
         result_image = self.postprocess_image(result[0][0], orig_im_size)
         if out_name != "":
             # if out_name is specified we save the image using that name