Update app.py

app.py

@@ -1,5 +1,6 @@
 # Import necessary libraries
 from together import Together
+import gradio as gr
 import os
 
 # Retrieve API key from Hugging Face Secrets
@@ -31,7 +32,7 @@ def generate_image(prompt, model="black-forest-labs/FLUX.1-schnell", steps=4):
 # Gradio Interface Setup
 with gr.Blocks() as app:
     gr.Markdown("## Real-Time AI Image Generator 🚀")
-    gr.Markdown("Start typing a prompt below to generate images in real-time using Together AI's FLUX model.")
+    gr.Markdown("Start typing a prompt below to generate images in real-time using FLUX Schnell model.")
 
     with gr.Row():
         prompt_box = gr.Textbox(label="Enter your prompt", placeholder="A horse on the moon")