Update app.py

app.py

@@ -1,55 +1,38 @@
-import torch
 import gradio as gr
-from diffusers import StableDiffusionPipeline
-
-model_id = "stabilityai/sd-turbo"
-device = "cuda" if torch.cuda.is_available() else "cpu"
-
-pipe = StableDiffusionPipeline.from_pretrained(
-    model_id,
-    torch_dtype=torch.float16 if device == "cuda" else torch.float32
-)
-pipe = pipe.to(device)
-
-styles = {
-    "Realistic": "highly detailed, realistic, 4k",
-    "Cartoon": "cartoon style, colorful",
-    "Anime": "anime style",
-    "Sketch": "pencil sketch"
+import requests
+
+HF_TOKEN = "images"
+
+API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
+
+headers = {
+    "Authorization": f"Bearer {HF_TOKEN}"
 }
 
-def generate_image(prompt, style, steps, guidance, seed):
+def generate_image(prompt):
     if not prompt.strip():
-        return None, "⚠️ Please enter a prompt."
+        return None, "Enter prompt"
 
-    styled_prompt = prompt + ", " + styles.get(style, "")
-    generator = torch.manual_seed(int(seed)) if seed != 0 else None
+    response = requests.post(
+        API_URL,
+        headers=headers,
+        json={"inputs": prompt}
+    )
 
-    image = pipe(
-        styled_prompt,
-        num_inference_steps=int(steps),
-        guidance_scale=float(guidance),
-        generator=generator
-    ).images[0]
+    if response.status_code != 200:
+        return None, response.text
 
-    return image, "✅ Done"
+    return response.content, "Done"
 
 with gr.Blocks() as app:
-    gr.Markdown("# 🎨 Text-to-Image Generator")
-    gr.Markdown("Powered by Stable Diffusion")
+    gr.Markdown("# Text to Image AI")
 
     prompt = gr.Textbox(label="Prompt")
-    style = gr.Dropdown(["Realistic", "Cartoon", "Anime", "Sketch"], value="Realistic")
-
-    steps = gr.Slider(10, 50, value=25)
-    guidance = gr.Slider(5.0, 15.0, value=7.5)
-    seed = gr.Number(value=0)
-
     btn = gr.Button("Generate")
 
-    image = gr.Image()
+    img = gr.Image(type="pil")
     status = gr.Textbox()
 
-    btn.click(generate_image, [prompt, style, steps, guidance, seed], [image, status])
+    btn.click(generate_image, prompt, [img, status])
 
 app.launch()