AYYasaswini commited on
Commit
0d99ad3
·
verified ·
1 Parent(s): c85b309

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -10
app.py CHANGED
@@ -308,16 +308,13 @@ def generate_with_prompt_style_guidance(prompt, style, seed=42):
308
 
309
  import gradio as gr
310
 
311
- dict_styles = {'Arcane':'styles/learned_embeds_arcane.bin',
312
- 'Button eyes':'styles/learned_embeds_buttoneyes.bin',
313
- 'Dr Strange': 'styles/learned_embeds_dr_strange.bin',
314
- 'GTA-5':'styles/learned_embeds_gta5.bin',
315
- 'Illustration': 'styles/learned_embeds_illustration.bin',
316
- 'Manga':'styles/learned_embeds_manga.bin',
317
- 'Matrix':'styles/learned_embeds_matrix.bin',
318
- 'Oil Painting':'styles/learned_embeds_oil.bin',
319
- 'Pokemon':'styles/learned_embeds_pokemon.bin',
320
- 'Stripes': 'styles/learned_embeds_stripe.bin'}
321
  # dict_styles.keys()
322
 
323
  def inference(prompt, style):
 
308
 
309
  import gradio as gr
310
 
311
+ #['<gartic-phone>', '<hawaiian shirt>','<gp>','<style-spdmn>','<yvmqznrm>']
312
+ dict_styles = {'gartic-phone':'styles/learned_embeds_gartic-phone.bin',
313
+ 'hawaiian shirt':'styles/learned_embeds_hawaiian-shirt.bin',
314
+ 'gp': 'styles/learned_embeds_phone01.bin',
315
+ 'style-spdmn':'styles/learned_embeds_style-spdmn.bin',
316
+ 'yvmqznrm': 'styles/learned_embedssd_yvmqznrm.bin',
317
+ }
 
 
 
318
  # dict_styles.keys()
319
 
320
  def inference(prompt, style):