Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ def resize(w_val,l_val,img):
20
 
21
  def infer(source_img, prompt, guide, steps, seed, Strength):
22
  generator = torch.Generator('cpu').manual_seed(seed)
23
- source_image = resize(768, 512, source_img)
24
  source_image.save('source.png')
25
  image_list = pipe([prompt], init_image=source_image, strength=Strength, guidance_scale=guide, num_inference_steps=steps)
26
  images = []
 
20
 
21
  def infer(source_img, prompt, guide, steps, seed, Strength):
22
  generator = torch.Generator('cpu').manual_seed(seed)
23
+ source_image = resize(512, source_img)
24
  source_image.save('source.png')
25
  image_list = pipe([prompt], init_image=source_image, strength=Strength, guidance_scale=guide, num_inference_steps=steps)
26
  images = []