Spaces:
Build error
Build error
thefreeham
commited on
Commit
•
ee8960e
1
Parent(s):
817095a
Update app.py
Browse files
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)
|