multimodalart HF staff commited on
Commit
a6c65a0
•
1 Parent(s): 388cd71

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ pipe = UnCLIPPipeline.from_pretrained("kakaobrain/karlo-v1-alpha", torch_dtype=t
6
  pipe = pipe.to("cuda")
7
 
8
  def run(prompt):
9
- images = pipe([prompt]*6).images
10
  return images
11
 
12
  css = '''
6
  pipe = pipe.to("cuda")
7
 
8
  def run(prompt):
9
+ images = pipe(prompt, num_images_per_prompt=6).images
10
  return images
11
 
12
  css = '''