Update app.py

app.py

@@ -40,7 +40,7 @@ def reduce_highlight(images):
     amber_error = (red_error + green_error + blue_error) / 3
     return amber_error
 
-def _inference(text, style, use_loss=False):
+def output(text, style, use_loss=False):
     if use_loss:
         image = gen_image_as_per_prompt(text, styles_dict[style], seed_dict[style], reduce_highlight)
     else:
@@ -49,22 +49,21 @@ def _inference(text, style, use_loss=False):
 
 
 title = "Stable Diffusion with different styles"
-description = "In this demo, the word 'puppy' is replaced by the style that is selected"
-examples = [["oil painting of a dragon in puppy style", "mosiac", True],
-            ["Spiderman in puppy style", "midjourney", True],
-            ["Batman in puppy style", "matrix", False],
-             ["Mojo Jojo in puppy style", "No style", False]]
+description = "Explore the versatility of artistic styles by transforming your prompts. This demo takes the word 'cat' in your input and applies a unique style of your choice, giving you a fresh visual interpretation."
+examples = [["A majestic lion with the playful expression of a puppy", "depthmap", True],
+            ["A futuristic robot designed in space suit", "midjourney", True],
+            ["A serene forest scene, with animals that resemble puppies", "cosmicgalaxy", False],
+             ["A warrior in an ancient battlefield, with a hint of puppy charm", "concept-art", False]]
 
 demo = gr.Interface(
-    _inference,
+    output,
     inputs=[
-        gr.Textbox(placeholder="Type a prompt with word 'puppy' in it..", container=False, scale=7),
+        gr.Textbox(placeholder="Explore the versatility of artistic styles by transforming your prompts. This demo takes the word 'cat' in your input and applies a unique style of your choice, giving you a fresh visual interpretation.", container=False, scale=7),
         gr.Radio(styles, label="Select a Style"),
         gr.Checkbox(label="Use custom loss")
     ],
     outputs=[
-        gr.Image(width=256, height=256, label="output")
-        # gr.Text(label="output")
+        gr.Image(width=512, height=512, label="output")
     ],
     title=title,
     description=description,