Update app.py

app.py

@@ -7,7 +7,8 @@ import torch
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 if torch.cuda.is_available():
-    pipe = DiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16", use_safetensors=True)
+    # pipe = DiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp32", use_safetensors=True)
+    pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, variant="fp32", use_safetensors=True)
     pipe.enable_xformers_memory_efficient_attention()
     pipe = pipe.to(device)
 else: 
@@ -36,10 +37,8 @@ def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance
     return image
 
 examples = [
-    "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k",
     "A playful Australian Shepherd dog running around in Central Park",
     "A tempting slice of Basque cheesecake",
-    "정글에 있는 우주 비행사, 차가운 색 팔레트, muted 색상, 상세한, 8k",
     "센트럴파크에서 뛰어 놀고 있는 오스트레일리언 셰퍼드 강아지 ",
     "맛있는 바스크 치즈케이크 조각",
 ]
@@ -131,7 +130,7 @@ with gr.Blocks(css=css) as demo:
                     minimum=1,
                     maximum=50,
                     step=1,
-                    value=20,  # 추론 단계를 20으로 설정하여 시간 단축
+                    value=10,  # 추론 단계를 10으로 설정하여 시간 단축
                 )
         
         gr.Examples(