Upload app.py

app.py

@@ -0,0 +1,27 @@
+import gradio as gr
+from diffusers import AutoPipelineForText2Image
+import torch
+
+# 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
+
+# Membuat Antarmuka Gradio
+demo = gr.Interface(
+    fn=generate_image,
+    inputs=gr.Textbox(label="Masukkan Prompt Anda"),
+    outputs=gr.Image(label="Hasil Gambar"),
+    title="AI Text-to-Image Generator",
+    description="Masukkan deskripsi teks untuk menghasilkan gambar menggunakan Stable Diffusion."
+)
+
+if __name__ == "__main__":
+    demo.launch()