Update app.py

app.py

@@ -9,13 +9,10 @@ css = """
 }
 """
 
-# Maximum image size
-MAX_IMAGE_SIZE = 1024
-
 # Load the ZB-Tech/Text-to-Image model
-def infer(prompt, height, width):
+def infer(prompt):
     model = gr.Interface.load("models/ZB-Tech/Text-to-Image")
-    return model(prompt, height=height, width=width)
+    return model(prompt)
 
 # Gradio app
 with gr.Blocks(css=css) as demo:
@@ -35,23 +32,6 @@ with gr.Blocks(css=css) as demo:
             lines=2,
         )
 
-        # Input: Height and Width sliders
-        height = gr.Slider(
-            label="Height",
-            minimum=256,
-            maximum=MAX_IMAGE_SIZE,
-            step=32,
-            value=1024,
-        )
-
-        width = gr.Slider(
-            label="Width",
-            minimum=256,
-            maximum=MAX_IMAGE_SIZE,
-            step=32,
-            value=1024,
-        )
-
         # Output: Generated Image
         result = gr.Image(label="Generated Image", type="pil")
 
@@ -59,7 +39,7 @@ with gr.Blocks(css=css) as demo:
         generate_button = gr.Button("Generate")
 
         # Button action: Call infer function
-        generate_button.click(infer, inputs=[prompt, height, width], outputs=result)
+        generate_button.click(infer, inputs=[prompt], outputs=result)
 
 # Launch the app
 if __name__ == "__main__":