rahul7star commited on
Commit
9571b1c
Β·
verified Β·
1 Parent(s): 2712575

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -10,11 +10,12 @@ from diffusers import DiffusionPipeline
10
 
11
  from diffusers import DiffusionPipeline
12
 
 
 
13
  pipe = DiffusionPipeline.from_pretrained("sarthak247/Wan2.1-T2V-1.3B-nf4")
14
 
15
  prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
16
  image = pipe(prompt).images[0]
17
-
18
  def generate_image(prompt):
19
  """Generates an image from text prompt using WAN 2.1"""
20
  image = pipe(prompt).images[0]
 
10
 
11
  from diffusers import DiffusionPipeline
12
 
13
+ from diffusers import DiffusionPipeline
14
+
15
  pipe = DiffusionPipeline.from_pretrained("sarthak247/Wan2.1-T2V-1.3B-nf4")
16
 
17
  prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
18
  image = pipe(prompt).images[0]
 
19
  def generate_image(prompt):
20
  """Generates an image from text prompt using WAN 2.1"""
21
  image = pipe(prompt).images[0]