Update app.py

app.py

@@ -8,10 +8,9 @@ transformers.logging.set_verbosity_error()
 
 # 모델 경로 목록
 model_paths = {
-    "sd_xl_base_1.0.safetensors": "runwayml/stable-diffusion-v1-5",
-    "arcane-diffusion-v3.ckpt": "nitrosocke/Arcane-Diffusion",
-    "Cyberpunk-Anime-Diffusion.ckpt": "nitrosocke/Cyberpunk-Anime-Diffusion",
-    # 다른 모델 경로도 여기에 추가할 수 있습니다.
+    "CompVis/stable-diffusion-v1-4": "CompVis/stable-diffusion-v1-4",
+    "runwayml/stable-diffusion-v1-5": "runwayml/stable-diffusion-v1-5",
+    "stabilityai/stable-diffusion-2-1": "stabilityai/stable-diffusion-2-1"
 }
 
 # VAE 경로
@@ -34,8 +33,7 @@ def generate_image(prompt, negative_prompt, model_name, steps, sampler, cfg_scal
         prompt=prompt,
         negative_prompt=negative_prompt,
         num_inference_steps=steps,
-        sampler=sampler,
-        cfg_scale=cfg_scale,
+        guidance_scale=cfg_scale,
         width=width,
         height=height,
         generator=generator,
@@ -57,7 +55,7 @@ with gr.Blocks(css=css) as demo:
         model_name = gr.Dropdown(
             label="Model (모델 선택)",
             choices=list(model_paths.keys()),
-            value="sd_xl_base_1.0.safetensors",
+            value="runwayml/stable-diffusion-v1-5",
             interactive=True
         )
 
@@ -83,8 +81,8 @@ with gr.Blocks(css=css) as demo:
                     with gr.Row():
                         sampler = gr.Dropdown(
                             label="Sampler (샘플러)",
-                            choices=["DDPM", "PNDM", "LMS"],
-                            value="DDPM"
+                            choices=["ddim", "ddpm", "pndm"],
+                            value="ddpm"
                         )
 
                     with gr.Row():
@@ -146,4 +144,3 @@ with gr.Blocks(css=css) as demo:
         )
 
 demo.launch()
-