Update app.py

app.py

@@ -8,27 +8,20 @@ def generate_image(text, seed):
     if seed is not None:
         random.seed(seed)
 
-    # Define slight variations of the original prompt for different poses/positions
-    variations = [
-        f"{text}, front view",
-        f"{text}, side view",
-        f"{text}, dynamic pose",
-        f"{text}, close-up view"
-    ]
+    if text in [example[0] for example in examples]:
+        print(f"Using example: {text}")
 
-    # Generate an image for each variation
-    images = [model(variation) for variation in variations]
-    return images
+    return model(text)
 
-# Define examples with pre-generated image previews
+# Examples for user guidance
 examples = [
-    ["Humanoid Cat Warrior, Full View", None, "path/to/humanoid_cat_preview.jpg"],
-    ["Warhammer Sisterhood", None, "path/to/warhammer_sisterhood_preview.jpg"],
-    ["Future Robots war", None, "path/to/future_robots_war_preview.jpg"],
-    ["Fantasy dragon", None, "path/to/fantasy_dragon_preview.jpg"]
+    ["Humanoid Cat Warrior, Full View", None],
+    ["Warhammer Sisterhood", None],
+    ["Future Robots war", None],
+    ["Fantasy dragon", None]
 ]
 
-# Custom CSS for styling
+# Custom CSS for gradient buttons and capsule shapes
 css = """
 /* General button styles */
 button {
@@ -68,20 +61,16 @@ input[type=range] {
 }
 """
 
-# Define a function to create the examples with preview images
-def create_example_row(example):
-    return [gr.Example(value=example[0], examples=None, label=example[0], image=example[2])]
-
-# Gradio interface
+# Gradio interface with custom styles
 interface = gr.Interface(
     fn=generate_image,
     inputs=[
         gr.Textbox(label="Type your imagination:", placeholder="Type or select an example..."),
         gr.Slider(minimum=0, maximum=10000, step=1, label="Seed (optional)")
     ],
-    outputs=[gr.Image(label=f"Generated Image {i+1}") for i in range(4)],
-    examples=create_example_row(examples),  # Use pre-generated images for previews
-    css=css,
+    outputs=gr.Image(label="Generated Image"),
+    examples=examples,
+    css=css,  # Apply custom CSS styling
     description="Note: The model is running on CPU; processing might take a bit longer. Thank you for your patience!"
 )