hshetty commited on
Commit
307831e
·
1 Parent(s): 25b84dd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ def generate(celebrity, setting):
30
  # ANSWER HERE
31
  prompt = 'A movie poster of {} in the movie{}'.format(celebrity,setting)
32
  latent_sample = torch.randn((1,4,64,64),generator = generator,device=device)
33
- gen_img = pipe(prompt,latents=latent_sample,num_inference_steps=100,guidance_scale=g_scale[i]).images[0]
34
  image = improve_image(gen_img)
35
  return image
36
 
 
30
  # ANSWER HERE
31
  prompt = 'A movie poster of {} in the movie{}'.format(celebrity,setting)
32
  latent_sample = torch.randn((1,4,64,64),generator = generator,device=device)
33
+ gen_img = pipe(prompt,latents=latent_sample,num_inference_steps=100,guidance_scale=20).images[0]
34
  image = improve_image(gen_img)
35
  return image
36