MaxMilan1
commited on
Commit
·
591bf9d
1
Parent(s):
f99c2d2
smol
Browse files
app.py
CHANGED
@@ -12,7 +12,7 @@ pipe.to("cuda")
|
|
12 |
@spaces.GPU
|
13 |
def generate_image(prompt, neg_prompt):
|
14 |
|
15 |
-
images = pipe(prompt
|
16 |
return images[0]
|
17 |
|
18 |
_TITLE = "Shoe Generator"
|
|
|
12 |
@spaces.GPU
|
13 |
def generate_image(prompt, neg_prompt):
|
14 |
|
15 |
+
images = pipe(prompt, negative_prompt=neg_prompt,)
|
16 |
return images[0]
|
17 |
|
18 |
_TITLE = "Shoe Generator"
|