hshetty commited on
Commit
8c3c143
·
1 Parent(s): 307831e

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=20).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=70,guidance_scale=14).images[0]
34
  image = improve_image(gen_img)
35
  return image
36