Zhibinhong
/

ReplacerRemover

Model card Files Files and versions

Zhibinhong commited on Apr 27, 2023

Commit

0639203

·

1 Parent(s): ac957a1

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -29,7 +29,7 @@ class EndpointHandler():
         if replacement=="":
             return self.model.inference_remove(raw_image,target)
         else:
-            return self.model.inference_replace_sam(raw_image,target,replacement)
 if __name__=="__main__":

         if replacement=="":
             return self.model.inference_remove(raw_image,target)
         else:
+            return json.dumps({'image':self.model.inference_replace_sam(raw_image,target,replacement)})
 if __name__=="__main__":