Update app.py

app.py

@@ -1,3 +1,21 @@
 import gradio as gr
 
-gr.load("models/stabilityai/stable-diffusion-3.5-large").launch()
+# Load the model
+model = gr.load("models/stabilityai/stable-diffusion-3.5-large")
+
+# Define a function to handle the model inference
+def generate_image(prompt, num_inference_steps):
+    return model(prompt=prompt, num_inference_steps=num_inference_steps)
+
+# Define the Gradio interface
+interface = gr.Interface(
+    fn=generate_image,  # The prediction function
+    inputs=[
+        gr.Textbox(label="Prompt"),  # Input for the text prompt
+        gr.Slider(minimum=1, maximum=100, default=50, label="Number of Inference Steps")  # Slider for steps
+    ],
+    outputs=gr.Image(label="Generated Image")  # Output component for the generated image
+)
+
+# Launch the interface
+interface.launch()