tonyassi commited on
Commit
39cb2e3
·
1 Parent(s): 57864f0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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,type):
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
- type = ' type:' + ','.join(type)
29
 
30
- prompt += description + color + features + occasion + type
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(