Spaces:

ifmain
/

Object_Remove

Running on Zero

ifmain commited on Mar 30

Commit

e8eb6dd

•

1 Parent(s): 71044b8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -133,7 +133,7 @@ def rmGPT(caption, remove_class):
 @spaces.GPU
 def ChangeOBJ(img1, response, mask1):
     size = img1.size
-    image = sdxl(prompt=response, image=img1, mask_image=mask1).images[0]
     return image.resize((size[0], size[1]))

 @spaces.GPU
 def ChangeOBJ(img1, response, mask1):
     size = img1.size
+    image = sdxl(prompt=response, image=img1, mask_image=mask1,strength=1.0).images[0]
     return image.resize((size[0], size[1]))