Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -13,7 +13,7 @@ pipe.unet.load_attn_procs(lora_model_path)
|
|
| 13 |
pipe.to("cuda")
|
| 14 |
|
| 15 |
|
| 16 |
-
def greet(description,color,features,occasion,
|
| 17 |
|
| 18 |
prompt = 'white background '
|
| 19 |
|
|
@@ -25,9 +25,9 @@ def greet(description,color,features,occasion,type):
|
|
| 25 |
|
| 26 |
occasion = ' occasion:' + ','.join(occasion)
|
| 27 |
|
| 28 |
-
|
| 29 |
|
| 30 |
-
prompt += description + color + features + occasion +
|
| 31 |
|
| 32 |
print('prompt:',prompt)
|
| 33 |
image = pipe(
|
|
|
|
| 13 |
pipe.to("cuda")
|
| 14 |
|
| 15 |
|
| 16 |
+
def greet(description,color,features,occasion,type_):
|
| 17 |
|
| 18 |
prompt = 'white background '
|
| 19 |
|
|
|
|
| 25 |
|
| 26 |
occasion = ' occasion:' + ','.join(occasion)
|
| 27 |
|
| 28 |
+
type_ = ' type:' + ','.join(type_)
|
| 29 |
|
| 30 |
+
prompt += description + color + features + occasion + type_
|
| 31 |
|
| 32 |
print('prompt:',prompt)
|
| 33 |
image = pipe(
|