Update app.py

app.py

@@ -1,27 +1,16 @@
-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()
+import os
+from huggingface_hub import InferenceClient
+
+client = InferenceClient(
+    provider="together",
+    api_key=os.environ["HF_TOKEN"],
+)
+
+# output is a PIL.Image object
+image = client.text_to_image(
+    "Astronaut riding a horse",
+    model="black-forest-labs/FLUX.1-dev",
+)
+
+if __name__ == "__main__":
+    demo.launch()