Update app.py

app.py

@@ -1,33 +1,24 @@
 import gradio as gr
-from diffusers import DiffusionPipeline
-import torch
+from transformers import pipeline
 
-import requests
+# Load Stable Diffusion model from Hugging Face
+generator = pipeline("text-to-image", model="stabilityai/stable-diffusion-3-medium-diffusers")
 
-API_URL = "https://router.huggingface.co/hf-inference/v1"
-headers = {"Authorization": "Bearer hf_xxxxxxxxxxxxxxxxxxxxxxxx"}
-
-def query(payload):
-    response = requests.post(API_URL, headers=headers, json=payload)
-    return response.content
-
-# Load the FLUX.1-dev model from Hugging Face
-pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.float16)
-pipe = pipe.to("cuda")
-
-# Define a function to generate an image based on the prompt
+# Define the chatbot function
 def generate_image(prompt: str):
-    # Generate the image using the pipeline
-    image = pipe(prompt).images[0]
+    # Generate the image from the prompt
+    image = generator(prompt)[0]["image"]
     return image
 
 # Create the Gradio interface
-iface = gr.Interface(fn=generate_image, 
-                     inputs="text", 
-                     outputs="image", 
-                     title="Text-to-Image with FLUX.1-dev",
-                     description="Enter a prompt to generate an image using the FLUX.1-dev model.")
+iface = gr.Interface(
+    fn=generate_image, 
+    inputs=gr.Textbox(label="Enter a description of the image"), 
+    outputs=gr.Image(label="Generated Image"),
+    live=True
+)
+
+# Launch the Gradio interface
+iface.launch(share=True)
 
-# Launch the app
-iface.launch()