Diogomc commited on
Commit
62a6ee1
·
verified ·
1 Parent(s): 3ec5e8d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,6 +1,6 @@
1
  from diffusers import DiffusionPipeline
2
 
3
- pipe = DiffusionPipeline.from_pretrained("Wan-AI/Wan2.1-I2V-14B-720P")
4
 
5
- prompt = "orange being cut"
6
- image = pipe(prompt).images[0]
 
1
  from diffusers import DiffusionPipeline
2
 
3
+ pipe = DiffusionPipeline.from_pretrained("ByteDance/AnimateDiff-Lightning")
4
 
5
+ prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
6
+ image = pipe(prompt).images[0]