Update app.py

app.py

@@ -1,32 +1,22 @@
 import gradio as gr
 
-# Load the model
-model = gr.load("models/black-forest-labs/FLUX.1-dev")
+def generate_image(prompt):
+    model = gr.load("models/black-forest-labs/FLUX.1-schnell")
+    return model(prompt)
 
-# Define the chatbot function
-def chatbot(message, history):
-    # Add the user's message to the history
-    history.append({"role": "user", "content": message})
+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.")
     
-    # Get the response from the model
-    response = model(message)
+    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")
     
-    # Add the model's response to the history
-    history.append({"role": "assistant", "content": response})
+    image_output = gr.Image(label="Generated Image")
     
-    return history
+    submit_button.click(fn=generate_image, inputs=text_input, outputs=image_output)
 
-# Create the Gradio interface
-with gr.Blocks(css="footer {visibility: hidden}") as demo:
-    gr.Markdown("# FLUX.1-dev Chatbot")
-    gr.Markdown("Welcome! This chatbot uses the FLUX.1-dev model. Feel free to ask anything!")
-    
-    chatbot = gr.Chatbot(height=400)
-    msg = gr.Textbox(label="Type your message here...")
-    clear = gr.Button("Clear")
-
-    msg.submit(chatbot, [msg, chatbot], [chatbot])
-    clear.click(lambda: None, None, chatbot, queue=False)
-
-# Launch the interface
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()