AYYasaswini commited on
Commit
fbf45f0
·
verified ·
1 Parent(s): 22dcc01

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -268,7 +268,7 @@ def generate_image_from_prompt(text_in, style_in):
268
 
269
  print(text_in)
270
  print(style_in)
271
- style_file = style_in
272
  idx = STYLE_LIST.index('learned_embeds_gartic-phone.bin')
273
  print(style_file)
274
  print(idx)
@@ -297,8 +297,8 @@ instructions = "Explore creative art generation using Stable Diffusion. Enter a
297
 
298
  demo = gr.Interface(generate_image_from_prompt,
299
  inputs = [gr.Textbox(label='Prompt'),
300
- gr.Dropdown(['<gartic-phone>', '<hawaiian shirt>','<gp>','<style-spdmn>','<yvmqznrm>'],
301
- value="<hawaiian shirt>",
302
  label=styles_label
303
  ),
304
  ],
 
268
 
269
  print(text_in)
270
  print(style_in)
271
+ style_file = style_in + '.bin'
272
  idx = STYLE_LIST.index('learned_embeds_gartic-phone.bin')
273
  print(style_file)
274
  print(idx)
 
297
 
298
  demo = gr.Interface(generate_image_from_prompt,
299
  inputs = [gr.Textbox(label='Prompt'),
300
+ gr.Dropdown(['learned_embeds_gartic-phone','learned_embeds_hawaiian-shirt', 'learned_embeds_phone0','learned_embeds_style-spdmn','learned_embedssd_yvmqznrm']'],
301
+ value="learned_embeds_gartic-phone",
302
  label=styles_label
303
  ),
304
  ],