Update app.py

app.py

@@ -1,15 +1,26 @@
-import os
-from huggingface_hub import InferenceClient
+import gradio as gr
+from diffusers import AutoPipelineForText2Image
+import torch
 
-client = InferenceClient(
-    provider="together",
-    api_key=os.environ["HF_TOKEN"],
-)
+# Memuat model text-to-image
+device = "cuda" if torch.cuda.is_available() else "cpu"
+pipeline = AutoPipelineForText2Image.from_pretrained(
+    "stable-diffusion-v1-5/stable-diffusion-v1-5", 
+    torch_dtype=torch.float16 if device == "cuda" else torch.float32
+).to(device)
+
+def generate_image(prompt):
+    # Menghasilkan gambar dari teks
+    image = pipeline(prompt).images[0]
+    return image
 
-# output is a PIL.Image object
-image = client.text_to_image(
-    "Astronaut riding a horse",
-    model="black-forest-labs/FLUX.1-dev",
+# Membuat Antarmuka Gradio
+demo = gr.Interface(
+    fn=generate_image,
+    inputs=gr.Textbox(label="Input your prompt"),
+    outputs=gr.Image(label="Image Result"),
+    title="AI Text-to-Image Generator",
+    description="Input Text via Stable Diffusion."
 )
 
 if __name__ == "__main__":