Spaces:
Running
on
Zero
Running
on
Zero
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(
|