tesT

README.md

@@ -1,5 +1,5 @@
 ---
-title: SSD 1B SDXL Refiner
+title: SSD-1B + SDXL 1.0 & Refiner
 emoji: 🏆
 colorFrom: indigo
 colorTo: pink

app.py

@@ -85,12 +85,19 @@ with gr.Blocks() as demo:
     with gr.Column():
         gr.Markdown("# Image Generation with SSD-1B")
         gr.Markdown("Enter a prompt and (optionally) a negative prompt to generate an image.")
+        
+        # Input fields for positive and negative prompts
         with gr.Row():
             prompt1 = gr.Textbox(label="Enter prompt")
             negative_prompt = gr.Textbox(label="Enter negative prompt (optional)")
-            generate_button1 = gr.Button("Generate Image")
-        output_image1 = gr.Image(type="filepath", label="Generated Image")
+        
+        # Button for generating the image
+        generate_button1 = gr.Button("Generate Image")
 
+        # Output image display, set to a larger default size
+        output_image1 = gr.Image(label="Generated Image")
+
+        # Click event for the generate button
         generate_button1.click(
             generate_and_save_image, 
             inputs=[prompt1, negative_prompt], 
@@ -100,16 +107,29 @@ with gr.Blocks() as demo:
     with gr.Column():
         gr.Markdown("## Refined Image Generation")
         gr.Markdown("Enter a prompt to generate a refined image.")
-        with gr.Row():
-            prompt2 = gr.Textbox(label="Enter prompt for refined generation")
-            generate_button2 = gr.Button("Generate Refined Image")
-        output_image2 = gr.Image(type="filepath", label="Generated Refined Image")
-
+        
+        # Input field for the prompt
+        prompt2 = gr.Textbox(label="Enter prompt for refined generation")
+        
+        # Button for generating the refined image
+        generate_button2 = gr.Button("Generate Refined Image")
+        
+        # Output refined image display, set to a larger default size
+        output_image2 = gr.Image(label="Generated Refined Image")
+
+        # Click event for the generate button
         generate_button2.click(
             generate_image_with_refinement, 
             inputs=[prompt2], 
             outputs=output_image2
         )
 
+    # Set the image display to be the largest element for both SSD-1B and refined generation
+    demo.update(
+        output_image1.style(width='100%', height='auto', min_height='400px'),
+        output_image2.style(width='100%', height='auto', min_height='400px')
+    )
+
+
 # Launch the combined Gradio app
 demo.launch()