Update app.py

app.py

@@ -1,22 +1,3 @@
 import gradio as gr
 
-def generate_image(prompt):
-    model = gr.load("models/black-forest-labs/FLUX.1-schnell")
-    return model(prompt)
-
-with gr.Blocks() as demo:
-    gr.Markdown("# Text to Image Generator")
-    gr.Markdown("Enter a text prompt to generate an image using the FLUX.1-dev model.")
-    
-    with gr.Row():
-        with gr.Column(scale=4):
-            text_input = gr.Textbox(label="Text Prompt", placeholder="Enter your prompt here...")
-        with gr.Column(scale=1):
-            submit_button = gr.Button("Generate Image")
-    
-    image_output = gr.Image(label="Generated Image")
-    
-    submit_button.click(fn=generate_image, inputs=text_input, outputs=image_output)
-
-if __name__ == "__main__":
-     .launch()
+gr.load("models/black-forest-labs/FLUX.1-schnell").launch()