ifmain commited on
Commit
d722d7e
1 Parent(s): 4a20809

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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, num_inference_steps=20, 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)
 
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)