Update app.py

app.py

@@ -54,18 +54,14 @@ def face_swap(dest_img):
     return final_image 
 
 
-def greet(description,color,features,occasion,type_):
-    
+def greet(description,color,features,occasion,type_,face):
+
+    # Parse input
     prompt = 'white background '
-    
     description = 'description:' + description.replace(' ', '-')
-
     color = ' color:' + ','.join(color)
-
     features = ' features:' + ','.join(features)
-
     occasion = ' occasion:' + ','.join(occasion)
-
     type_ = ' type:' + ','.join(type_)
    
     prompt += description + color + features +  occasion + type_
@@ -91,6 +87,7 @@ iface = gr.Interface(fn=greet,
                             gr.Dropdown(label='Features',choices=['3/4-sleeve','Babydoll','Closed-Back','Corset','Crochet','Cutouts','Draped','Floral','Gloves','Halter','Lace','Long','Long-Sleeve','Midi','No-Slit','Off-The-Shoulder','One-Shoulder','Open-Back','Pockets','Print','Puff-Sleeve','Ruched','Satin','Sequins','Shimmer','Short','Short-Sleeve','Side-Slit','Square-Neck','Strapless','Sweetheart-Neck','Tight','V-Neck','Velvet','Wrap'],multiselect=True),
                             gr.Dropdown(label='Occasion',choices=['Homecoming','Casual','Wedding-Guest','Festival','Sorority','Day','Vacation','Summer','Pool-Party','Birthday','Date-Night','Party','Holiday','Winter-Formal','Valentines-Day','Prom','Graduation'],multiselect=True),
                             gr.Dropdown(label='Type',choices=['Mini-Dresses','Midi-Dresses','Maxi-Dresses','Two-Piece-Sets','Rompers','Jeans','Jumpsuits','Pants','Tops','Jumpers/Cardigans','Skirts','Shorts','Bodysuits','Swimwear'],multiselect=True),
+                            gr.Dropdown(label='Face',choices=['Normal','Cat','Lisa','Mila']),
                             ], 
                      outputs="image")
 iface.launch()