Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -133,13 +133,13 @@ def rmGPT(caption, remove_class):
|
|
133 |
@spaces.GPU
|
134 |
def ChangeOBJ(img1, response, mask1):
|
135 |
size = img1.size
|
136 |
-
image = sdxl(prompt=response, image=img1, mask_image=mask1,guidance_scale=7.0,
|
137 |
return image.resize((size[0], size[1]))
|
138 |
|
139 |
|
140 |
def full_pipeline(image, target):
|
141 |
img1 = Image.fromarray(image.astype('uint8'), 'RGB')
|
142 |
-
img1 = img1.resize((512, 512))
|
143 |
allMask = getSegments(img1)
|
144 |
tartget_to_remove = get_most_similar_string(target, list(allMask.keys()))
|
145 |
caption = getDescript(image_captioner, img1)
|
|
|
133 |
@spaces.GPU
|
134 |
def ChangeOBJ(img1, response, mask1):
|
135 |
size = img1.size
|
136 |
+
image = sdxl(prompt=response, image=img1, mask_image=mask1,guidance_scale=7.0, strength=0.99).images[0]
|
137 |
return image.resize((size[0], size[1]))
|
138 |
|
139 |
|
140 |
def full_pipeline(image, target):
|
141 |
img1 = Image.fromarray(image.astype('uint8'), 'RGB')
|
142 |
+
#img1 = img1.resize((512, 512))
|
143 |
allMask = getSegments(img1)
|
144 |
tartget_to_remove = get_most_similar_string(target, list(allMask.keys()))
|
145 |
caption = getDescript(image_captioner, img1)
|