MaxMilan1
commited on
Commit
·
70ea20b
1
Parent(s):
fb01197
remove output type (PIL)
Browse files
app.py
CHANGED
|
@@ -18,7 +18,7 @@ pipe.to("cuda")
|
|
| 18 |
@spaces.GPU
|
| 19 |
def generate_image(prompt, neg_prompt):
|
| 20 |
prompt += "no background, side view, minimalist shot"
|
| 21 |
-
image = pipe(prompt, negative_prompt=neg_prompt
|
| 22 |
image2 = rembg.remove(image)
|
| 23 |
return image, image2
|
| 24 |
|
|
|
|
| 18 |
@spaces.GPU
|
| 19 |
def generate_image(prompt, neg_prompt):
|
| 20 |
prompt += "no background, side view, minimalist shot"
|
| 21 |
+
image = pipe(prompt, negative_prompt=neg_prompt)
|
| 22 |
image2 = rembg.remove(image)
|
| 23 |
return image, image2
|
| 24 |
|