Update app.py

app.py

@@ -1,46 +1,23 @@
-import requests
-import base64
-from PIL import Image
-from io import BytesIO
 import gradio as gr
+from diffusers import StableDiffusionPipeline
+import torch
 
-API_KEY = "sk-ESkTZ28wTlIcedRXuq2KTYbgunw578RkXprkewhJpMsu69Ec"  # REPLACE THIS!
+model_id = "CompVis/stable-diffusion-v1-4"
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+pipe = pipe.to("cuda")
 
 def generate_image(prompt):
-    # Step 1: Make the API request
-    response = requests.post(
-        "https://api.stability.ai/v1/generation/stable-diffusion-v1-5/text-to-image",
-        headers={
-            "Authorization": f"Bearer {API_KEY}",
-            "Content-Type": "application/json"
-        },
-        json={
-            "text_prompts": [{"text": prompt}],
-            "cfg_scale": 7,
-            "height": 512,
-            "width": 512,
-            "samples": 1,
-            "steps": 30
-        }
-    )
-
-    # Step 2: Error check
-    if response.status_code != 200:
-        return f"Error: {response.status_code} - {response.text}"
-
-    # Step 3: Convert base64 to PIL.Image
-    image_data = response.json()["artifacts"][0]["base64"]
-    image = Image.open(BytesIO(base64.b64decode(image_data))).convert("RGB")  # Ensure RGB
-
+    image = pipe(prompt).images[0]
     return image
 
-# Step 4: Gradio Interface — PIL Image Return Only
-demo = gr.Interface(
+iface = gr.Interface(
     fn=generate_image,
-    inputs=gr.Textbox(label="Enter a prompt"),
+    inputs=gr.Textbox(label="prompt", placeholder="Enter your prompt here..."),
     outputs=gr.Image(type="pil", label="Generated Image"),
-    title="✨ Speech2Image Generator ✨"
+    title="Speech2Image Generator (Fast API Powered)",
+    theme="default",
 )
 
-demo.launch()
+iface.launch()
+