MaxMilan1 commited on
Commit
f99c2d2
·
1 Parent(s): 12863e1
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ pipe.to("cuda")
10
 
11
  # Function to generate an image from text using diffusion
12
  @spaces.GPU
13
- def generate_image(prompt):
14
 
15
  images = pipe(prompt=prompt, negative_prompt=neg_prompt,)
16
  return images[0]
 
10
 
11
  # Function to generate an image from text using diffusion
12
  @spaces.GPU
13
+ def generate_image(prompt, neg_prompt):
14
 
15
  images = pipe(prompt=prompt, negative_prompt=neg_prompt,)
16
  return images[0]