thefreeham commited on
Commit
ee8960e
1 Parent(s): 817095a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -12,7 +12,8 @@ ldm = DiffusionPipeline.from_pretrained("fusing/latent-diffusion-text2im-large")
12
  generator = torch.manual_seed(42)
13
 
14
  def greet(name):
15
- prompt = "A squirrel eating a burger"
 
16
  image = ldm([prompt], generator=generator, eta=0.3, guidance_scale=6.0, num_inference_steps=50)
17
 
18
  image_processed = image.cpu().permute(0, 2, 3, 1)
12
  generator = torch.manual_seed(42)
13
 
14
  def greet(name):
15
+ #prompt = "A squirrel eating a burger"
16
+ prompt = name
17
  image = ldm([prompt], generator=generator, eta=0.3, guidance_scale=6.0, num_inference_steps=50)
18
 
19
  image_processed = image.cpu().permute(0, 2, 3, 1)