Update app.py

app.py

@@ -263,13 +263,13 @@ def generate_loss_based_image(text_embeddings, seed, max_length):
 
 
 def generate_image_from_prompt(text_in, style_in):
-    STYLE_LIST = ['learned_embeds_gartic-phone.bin','learned_embeds_hawaiian-shirt.bin', 'learned_embeds_phone0.bin1','learned_embeds_style-spdmn.bin','learned_embedssd_yvmqznrm.bin']
+    STYLE_LIST = ['oil_style.bin', 'valorant_style.bin', 'cartoon_syle.bin', 'space_style.bin', 'terraria_syle.bin']
     STYLE_SEEDS = [128, 64, 128, 64, 128]
     
     print(text_in)
     print(style_in)
-    style_file = style_in + '.bin'
-    idx = STYLE_LIST.index('learned_embeds_gartic-phone.bin')
+    style_file = style_in + '_style.bin'
+    idx = STYLE_LIST.index(style_file)
     print(style_file)
     print(idx)  
     
@@ -290,26 +290,28 @@ def generate_image_from_prompt(text_in, style_in):
 title = 'Stable Diffusion Art Generator'
 
 # Add clear and concise labels and instructions
-prompt_label = "A campfire (oil on canvas)"
+prompt_label = "Enter a prompt (e.g., 'man in style of puppy'):"
 styles_label = "Select a Pretrained Style:"
 
 instructions = "Explore creative art generation using Stable Diffusion. Enter a prompt and choose a style to get started."
 
 demo = gr.Interface(generate_image_from_prompt,
-                    inputs = [gr.Textbox(label='Prompt'), 
-                              gr.Dropdown(['learned_embeds_gartic-phone','learned_embeds_hawaiian-shirt', 'learned_embeds_phone0','learned_embeds_style-spdmn','learned_embedssd_yvmqznrm'],
-                              value="learned_embeds_gartic-phone",
-                            label=styles_label
+                    inputs=[
+                        gr.Textbox('man in style of puppy', label=prompt_label, style="width: 100%"),
+                        gr.Dropdown(
+                            ['oil', 'valorant', 'cartoon', 'space', 'terraria'],
+                            value="valorant",
+                            label=styles_label,
+                            style="width: 100%"
                         ),
                     ],
                     outputs=[
                         gr.Gallery(label="Generated Images", show_label=False, elem_id="gallery", columns=[2], rows=[2],
-                                   object_fit="contain"),
+                                   object_fit="contain", style="max-width: 800px; margin: 20px auto;"),
                     ],
                     title=title,
                     description=instructions,
                     theme="compact"  # Apply a predefined theme
                    )
-                    
 
 demo.launch(debug=True)