Upload app.py

app.py

@@ -285,7 +285,10 @@ 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', 
+    'Pokemon':'styles/learned_embeds_pokemon.bin',
+    'Illustration': 'styles/learned_embeds_illustration.bin', 
+    'Matrix':'styles/learned_embeds_matrix.bin', 
+    'Oil Painting':'styles/learned_embeds_oil.bin',  
 }
 # dict_styles.keys()
 
@@ -301,13 +304,10 @@ def inference(prompt, style,seed):
 
 title = "Stable Diffusion and Textual Inversion"
 description = "A simple Gradio interface to stylize Stable Diffusion outputs"
-examples = examples = [['Pink Ferrari Car', '24041975', 'Manga'], 
-            ['A man sipping tea wearing a spacesuit on the moon', '24041975', 'GTA-5']] 
-            # Example inputs
-
+examples = [["Pink Ferrari Car", 24041975,"Manga"], ["A man sipping tea wearing a spacesuit on the moon",24041975, "GTA-5"]]  # Added valid styles
 
 demo = gr.Interface(inference,
-                    inputs = [gr.Textbox(label='Prompt'),    gr.Textbox(label='Seed', value='24041975'),                           
+                    inputs = [gr.Textbox(label='Prompt', value='Pink Ferrari Car'),    gr.Textbox(label='Seed', value='24041975'),                           
                               gr.Dropdown(['Dr Strange', 'GTA-5', 'Manga', 'Pokemon'], label='Style')],
                               
                     outputs = [
@@ -315,7 +315,7 @@ demo = gr.Interface(inference,
                               ],
                     title = title,
                     description = description,
-                    examples = examples, 
+                    # examples = examples, 
                     # cache_examples=True
                     )
 demo.launch()