Update app.py

app.py

@@ -43,21 +43,13 @@ def infer(prompt, model_size, seed=42, randomize_seed=False, width=1024, height=
     print(img)
     return img.images[0], seed
     
-# Different examples for each model size
-examples_06B = [
-    "a majestic castle on a floating island",
-    "a robotic chef cooking in a futuristic kitchen",
-    "a magical forest with glowing mushrooms"
+examples = [
+    ["a tiny astronaut hatching from an egg on the moon", "0.6B"],
+    ["a cat holding a sign that says hello world", "1.6B"],
+    ["an anime illustration of a wiener schnitzel", "0.6B"],
+    ["a photorealistic landscape of mountains at sunset", "1.6B"],
 ]
 
-examples_16B = [
-    "a steampunk city with airships in the sky",
-    "a photorealistic fox in a snowy landscape",
-    "an underwater temple with ancient ruins"
-]
-
-# We'll use the appropriate set based on the model selection
-
 css="""
 #col-container {
     margin: 0 auto;
@@ -140,24 +132,13 @@ with gr.Blocks(css=css) as demo:
                     value=2,
                 )
         
-        with gr.Row():
-            examples_container = gr.Examples(
-                examples = examples_06B,  # Start with 0.6B examples
-                fn = infer,
-                inputs = [prompt, model_size],
-                outputs = [result, seed],
-                cache_examples="lazy",
-                label="Example Prompts"
-            )
-            
-        # Update examples when model size changes
-        def update_examples(model_choice):
-            if model_choice == "0.6B":
-                return gr.Examples.update(examples=examples_06B)
-            else:
-                return gr.Examples.update(examples=examples_16B)
-                
-        model_size.change(fn=update_examples, inputs=[model_size], outputs=[examples_container])
+        gr.Examples(
+            examples = examples,
+            fn = infer,
+            inputs = [prompt, model_size],  # Add model_size to inputs
+            outputs = [result, seed],
+            cache_examples="lazy"
+        )
 
     gr.on(
         triggers=[run_button.click, prompt.submit],