Create app.py

app.py

@@ -0,0 +1,39 @@
+import gradio as gr
+from diffusers import StableDiffusionPipeline
+import torch
+
+# Tải mô hình Stable Diffusion
+model_id = "runwayml/stable-diffusion-v1-5"
+pipe = StableDiffusionPipeline.from_pretrained(
+    model_id, 
+    torch_dtype=torch.float16,
+    use_auth_token=False  # Nếu dùng mô hình riêng, thêm token từ Hugging Face
+)
+pipe = pipe.to("cuda")  # Chuyển sang GPU (nếu Space có GPU)
+
+# Hàm tạo hình ảnh
+def generate_image(prompt, negative_prompt="", num_inference_steps=50, guidance_scale=7.5):
+    image = pipe(
+        prompt,
+        negative_prompt=negative_prompt,
+        num_inference_steps=num_inference_steps,
+        guidance_scale=guidance_scale
+    ).images[0]
+    return image
+
+# Tạo giao diện Gradio
+interface = gr.Interface(
+    fn=generate_image,
+    inputs=[
+        gr.Textbox(label="Prompt", placeholder="Enter your prompt here, e.g., 'A futuristic city at sunset'"),
+        gr.Textbox(label="Negative Prompt (optional)", placeholder="Things to avoid, e.g., 'blurry, low quality'"),
+        gr.Slider(label="Inference Steps", minimum=10, maximum=100, value=50, step=1),
+        gr.Slider(label="Guidance Scale", minimum=1, maximum=20, value=7.5, step=0.5)
+    ],
+    outputs=gr.Image(label="Generated Image"),
+    title="Text-to-Image with Stable Diffusion",
+    description="Enter a prompt to generate an image using Stable Diffusion."
+)
+
+# Khởi chạy giao diện
+interface.launch(server_name="0.0.0.0", server_port=7860)