Update app.py

app.py

@@ -285,16 +285,12 @@ def generate_with_prompt_style_guidance(prompt, style, seed=42):
 
 import gradio as gr
 
-dict_styles = {'Arcane':'styles/learned_embeds_arcane.bin', 
-               'Button eyes':'styles/learned_embeds_buttoneyes.bin', 
-               'Dr Strange': 'styles/learned_embeds_dr_strange.bin', 
-               'GTA-5':'styles/learned_embeds_gta5.bin', 
-               'Illustration': 'styles/learned_embeds_illustration.bin', 
-               'Manga':'styles/learned_embeds_manga.bin', 
-               'Matrix':'styles/learned_embeds_matrix.bin', 
-               'Oil Painting':'styles/learned_embeds_oil.bin', 
-               'Pokemon':'styles/learned_embeds_pokemon.bin', 
-               'Stripes': 'styles/learned_embeds_stripe.bin'}
+dict_styles = { 
+    'Dr Strange': 'styles/learned_embeds_dr_strange.bin', 
+    'GTA-5':'styles/learned_embeds_gta5.bin', 
+    'Manga':'styles/learned_embeds_manga.bin', 
+    'Pokemon':'styles/learned_embeds_pokemon.bin', 
+}
 # dict_styles.keys()
 
 def inference(prompt, style): 
@@ -312,8 +308,8 @@ examples = [['A man sipping wine wearing a spacesuit on the moon', 'Stripes']]
 
 demo = gr.Interface(inference,
                     inputs = [gr.Textbox(label='Prompt'), 
-                              gr.Dropdown(['Arcane', 'Button eyes', 'Dr Strange', 'GTA-5', 'Illustration', 
-                                           'Manga', 'Matrix', 'Oil Painting', 'Pokemon', 'Stripes'], label='Style')
+                              gr.Dropdown(['Dr Strange', 'GTA-5', 
+                                           'Manga', 'Pokemon'], label='Style')
                               ],
                     outputs = [
                               gr.Image(label="Stable Diffusion Output"),