jepz commited on
Commit
cb41890
·
verified ·
1 Parent(s): 863fa80

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -42,13 +42,13 @@ def infer(init_img, prompt, negative_prompt, seed, randomize_seed, width, height
42
  if randomize_seed:
43
  seed = random.randint(0, MAX_SEED)
44
 
45
- init_img = Image.open(init_img).convert("RGB")
46
- init_image.thumbnail((768, 768))
47
 
48
  generator = torch.Generator().manual_seed(seed)
49
 
50
  image = pipe(
51
- image = init_img,
52
  prompt = prompt,
53
  negative_prompt = negative_prompt,
54
  guidance_scale = guidance_scale,
@@ -166,7 +166,7 @@ with gr.Blocks(css=css) as demo:
166
 
167
  run_button.click(
168
  fn = infer,
169
- inputs = [init_img.convert("RGB"), prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, strength],
170
  outputs = [result]
171
  )
172
 
 
42
  if randomize_seed:
43
  seed = random.randint(0, MAX_SEED)
44
 
45
+ uploaded_img = Image.open(init_img).convert("RGB")
46
+ uploaded_img.thumbnail((768, 768))
47
 
48
  generator = torch.Generator().manual_seed(seed)
49
 
50
  image = pipe(
51
+ image = uploaded_img,
52
  prompt = prompt,
53
  negative_prompt = negative_prompt,
54
  guidance_scale = guidance_scale,
 
166
 
167
  run_button.click(
168
  fn = infer,
169
+ inputs = [init_img, prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, strength],
170
  outputs = [result]
171
  )
172