Update app.py

app.py

@@ -9,7 +9,7 @@ from translatepy import Translator
 
 translator = Translator()
 
-# Constants
+# 상수 정의
 model = "Corcelio/mobius"
 vae_model = "madebyollin/sdxl-vae-fp16-fix"
 
@@ -29,20 +29,19 @@ JS = """function () {
   }
 }"""
 
-# Load VAE component
+# VAE 컴포넌트 로드
 vae = AutoencoderKL.from_pretrained(
     vae_model, 
     torch_dtype=torch.float16
 )
 
-# Ensure model and scheduler are initialized in GPU-enabled function
+# GPU 사용 가능한 경우 모델 및 스케줄러 초기화
 if torch.cuda.is_available():
     pipe = StableDiffusionXLPipeline.from_pretrained(model, vae=vae, torch_dtype=torch.float16).to("cuda")
 
 pipe.scheduler = KDPM2AncestralDiscreteScheduler.from_config(pipe.scheduler.config)
 
-
-# Function 
+# 함수 정의
 @spaces.GPU()
 def generate_image(
     prompt,
@@ -50,23 +49,21 @@ def generate_image(
     width=1024,
     height=1024,
     scale=1.5,
-    steps=30,
-    clip=3):
+    steps=30):
     
     prompt = str(translator.translate(prompt, 'English'))
 
     print(f'prompt:{prompt}')
     
-    image = pipe(
+    images = pipe(
         prompt, 
         negative_prompt=negative, 
         width=width,
         height=height,
         guidance_scale=scale,
         num_inference_steps=steps,
-        clip_skip=clip,
-    )
-    return image.images[0], image.images[1]  # Return two images
+    ).images
+    return images[0], images[1]  # 두 이미지를 반환
 
 
 examples = [
@@ -83,7 +80,7 @@ examples = [
 ]
 
 
-# Gradio Interface
+# Gradio 인터페이스
 
 with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
     gr.HTML("<h1><center>나만의 모델 캐릭터 생성</center></h1>")
@@ -126,13 +123,6 @@ with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
                 step=1,
                 value=50,
             )
-            clip = gr.Slider(
-                label="Clip Skip",
-                minimum=1,
-                maximum=10,
-                step=1,
-                value=3,
-            )      
     gr.Examples(
         examples=examples,
         inputs=prompt,
@@ -142,11 +132,11 @@ with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
     )
 
     prompt.submit(fn=generate_image,
-                 inputs=[prompt, negative, width, height, scale, steps, clip],
+                 inputs=[prompt, negative, width, height, scale, steps],
                  outputs=[img1, img2],
                  )
     submit.click(fn=generate_image,
-                 inputs=[prompt, negative, width, height, scale, steps, clip],
+                 inputs=[prompt, negative, width, height, scale, steps],
                  outputs=[img1, img2],
                  )