Update app.py

app.py

@@ -152,8 +152,8 @@ def ref_loss(images,ref_image):
 
 def inference(prompt, style_index):
 
-    styles = ['<midjourney-style>', '<hitokomoru-style>','<birb-style>','<summie-style>','<illustration-style>','<m-geo>']
-    embed = ['learned_embeds_m.bin','learned_embeds_h.bin', 'learned_embeds.bin',   'learned_embeds_s.bin','learned_embeds_i.bin','learned_embeds_mg.bin']
+    styles = ['<midjourney-style>', '<hitokomoru-style>','<birb-style>','<summie-style>','<illustration-style>','<m-geo>','<buhu>']
+    embed = ['learned_embeds_m.bin','learned_embeds_h.bin', 'learned_embeds.bin',   'learned_embeds_s.bin','learned_embeds_i.bin','learned_embeds_mg.bin','learned_embeds_buhu']
 
 
     # Tokenize
@@ -329,9 +329,9 @@ gallery = gr.Gallery(label="Generated images", show_label=True, elem_id="gallery
 gr.Interface(fn=inference, inputs=["text",
 
     gr.Radio([('<midjourney-style>',0), ('<hitokomoru-style>',1),('<birb-style>',2),
-              ('<summie-style>',3),('<illustration-style>',4),('<m-geo>',5)] , value = 0, label = 'Styles')],
+              ('<summie-style>',3),('<illustration-style>',4),('<m-geo>',5),('<buhu>',6)] , value = 0, label = 'Please select your choice of Style')],
     outputs = gallery, title = title,
 examples = [['a girl playing in snow',0],
-                ['an oil painting of a goddess',1],
+                ['an oil painting of a goddess',6],
                 ['a rabbit on the moon', 5 ]], ).launch(debug=True)