Update app.py

app.py

@@ -1,57 +1,13 @@
 from diffusers import StableDiffusionPipeline
-from huggingface_hub import hf_hub_download
 import torch
-import gradio as gr
-
-# 模型和 LoRA 存儲庫信息
-base_model = "stabilityai/stable-diffusion-xl-base-1.0"  # 基礎模型
-repo_id = "hyder133/chiikawa_stype"                     # Hugging Face 存儲庫 ID
-lora_filename = "tkw1.safetensors"                      # LoRA 權重文件名
-
-# 嘗試加載到 GPU，如果不可用則回退到 CPU
-device = "cuda" if torch.cuda.is_available() else "cpu"
-dtype = torch.float16 if device == "cuda" else torch.float32
 
 # 加載基礎模型
-pipe = StableDiffusionPipeline.from_pretrained(base_model, torch_dtype=dtype)
-pipe = pipe.to(device)
-
-# 下載並加載 LoRA 權重
-lora_file_path = hf_hub_download(repo_id=repo_id, filename=lora_filename)
-pipe.load_lora_weights(lora_file_path)
-
-# 測試生成函數
-def generate_image(prompt, width, height, steps, guidance_scale):
-    # 使用模型生成圖像
-    image = pipe(
-        prompt,
-        width=width,
-        height=height,
-        num_inference_steps=steps,
-        guidance_scale=guidance_scale
-    ).images[0]
-    return image
-
-# Gradio 界面
-with gr.Blocks() as demo:
-    gr.Markdown("# Chiikawa Style Image Generator")
-    with gr.Row():
-        prompt = gr.Textbox(label="Prompt", placeholder="Enter your text prompt")
-    with gr.Row():
-        width = gr.Slider(256, 1024, value=512, step=64, label="Width")
-        height = gr.Slider(256, 1024, value=512, step=64, label="Height")
-    with gr.Row():
-        steps = gr.Slider(10, 50, value=30, step=5, label="Steps")
-        guidance_scale = gr.Slider(1, 20, value=7.5, step=0.5, label="Guidance Scale")
-    with gr.Row():
-        generate_button = gr.Button("Generate")
-    output = gr.Image(label="Generated Image")
-
-    generate_button.click(
-        generate_image,
-        inputs=[prompt, width, height, steps, guidance_scale],
-        outputs=output
-    )
-
-# 啟動應用
-demo.launch()
+pipe = StableDiffusionPipeline.from_pretrained(
+    "stabilityai/stable-diffusion-xl-base-1.0", 
+    torch_dtype=torch.float16
+).to("cuda")
+
+# 測試生成圖像
+prompt = "A cute cartoon character in Chiikawa style, ultra-detailed"
+image = pipe(prompt).images[0]
+image.save("output.png")