Update app.py

app.py

@@ -5,7 +5,7 @@ import os
 from compel import Compel, ReturnedEmbeddingsType
 from diffusers import DiffusionPipeline
 
-# 모델 로드
+# Load model
 model_name = os.environ.get('MODEL_NAME', 'UnfilteredAI/NSFW-gen-v2')
 pipe = DiffusionPipeline.from_pretrained(
     model_name,
@@ -13,7 +13,7 @@ pipe = DiffusionPipeline.from_pretrained(
 )
 pipe.to('cuda')
 
-# Compel 설정
+# Compel setup
 compel = Compel(
   tokenizer=[pipe.tokenizer, pipe.tokenizer_2],
   text_encoder=[pipe.text_encoder, pipe.text_encoder_2],
@@ -21,10 +21,10 @@ compel = Compel(
   requires_pooled=[False, True]
 )
 
-# 기본 네거티브 프롬프트
+# Default negative prompt
 default_negative_prompt = "(low quality, worst quality:1.2), very displeasing, 3d, watermark, signature, ugly, poorly drawn, (deformed | distorted | disfigured:1.3), bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, mutated hands and fingers:1.4, disconnected limbs, blurry, amputation."
 
-# 예제 프롬프트 목록
+# Example prompts
 example_prompts = [
     ["a beautiful woman in a summer dress at the beach, golden sunset, professional photography, 8k", default_negative_prompt, 40, 7.5, 1024, 1024, 4],
     ["portrait of a cyberpunk character, highly detailed, neon lights, futuristic cityscape in background, 8k, ultra realistic", default_negative_prompt, 50, 8.0, 768, 1024, 4],
@@ -32,7 +32,7 @@ example_prompts = [
     ["photorealistic portrait of a stunning model, studio lighting, fashion photography", default_negative_prompt, 45, 7.0, 1024, 1024, 4],
 ]
 
-# 이미지 생성 함수
+# Image generation function
 @spaces.GPU(duration=120)
 def generate(prompt, negative_prompt, num_inference_steps, guidance_scale, width, height, num_samples, progress=gr.Progress()):
     progress(0, desc="Preparing")
@@ -50,12 +50,13 @@ def generate(prompt, negative_prompt, num_inference_steps, guidance_scale, width
         width=width,
         height=height,
         num_images_per_prompt=num_samples,
-        callback=lambda i, t, latents: progress((i + 1) / num_inference_steps)
+        callback_steps=1,  # Fixed: Added callback_steps parameter
+        callback_on_step_end=lambda i, t, latents: progress((i + 1) / num_inference_steps)
     ).images
     
     return images
 
-# CSS 스타일
+# CSS styles
 css = """
 .gallery-item {
     transition: transform 0.2s;
@@ -101,21 +102,21 @@ css = """
 }
 """
 
-# Gradio 인터페이스
+# Gradio interface
 with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
     gr.HTML("""
     <div class="header">
-        <h1>🎨 UnfilteredAI NSFW-gen-v2 이미지 생성기</h1>
-        <p>창의적인 프롬프트를 입력하고 고품질 이미지를 생성해보세요.</p>
+        <h1>🎨 UnfilteredAI NSFW-gen-v2 Image Generator</h1>
+        <p>Enter creative prompts and generate high-quality images.</p>
     </div>
     """)
     
     with gr.Row():
         with gr.Column(scale=2):
             with gr.Group(elem_classes="prompt-container"):
-                prompt = gr.Textbox(label="프롬프트", placeholder="원하는 이미지를 설명해주세요...", lines=3)
+                prompt = gr.Textbox(label="Prompt", placeholder="Describe your desired image...", lines=3)
                 negative_prompt = gr.Textbox(
-                    label="네거티브 프롬프트", 
+                    label="Negative Prompt", 
                     value=default_negative_prompt,
                     lines=3
                 )
@@ -123,22 +124,22 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
             with gr.Group(elem_classes="slider-container"):
                 with gr.Row():
                     with gr.Column():
-                        steps = gr.Slider(minimum=20, maximum=100, value=60, step=1, label="추론 단계 (Quality)", info="높을수록 품질 향상 (시간 증가)")
-                        guidance = gr.Slider(minimum=1, maximum=15, value=7, step=0.1, label="가이던스 스케일 (Creativity)", info="낮을수록 더 창의적인 결과")
+                        steps = gr.Slider(minimum=20, maximum=100, value=60, step=1, label="Inference Steps (Quality)", info="Higher values improve quality (longer generation time)")
+                        guidance = gr.Slider(minimum=1, maximum=15, value=7, step=0.1, label="Guidance Scale (Creativity)", info="Lower values create more creative results")
                     
                     with gr.Column():
                         with gr.Row():
-                            width = gr.Slider(minimum=512, maximum=1536, value=1024, step=128, label="너비")
-                            height = gr.Slider(minimum=512, maximum=1536, value=1024, step=128, label="높이")
+                            width = gr.Slider(minimum=512, maximum=1536, value=1024, step=128, label="Width")
+                            height = gr.Slider(minimum=512, maximum=1536, value=1024, step=128, label="Height")
                         
-                        num_samples = gr.Slider(minimum=1, maximum=8, value=4, step=1, label="생성할 이미지 수", info="한 ��에 생성할 이미지 수")
+                        num_samples = gr.Slider(minimum=1, maximum=8, value=4, step=1, label="Number of Images", info="Number of images to generate at once")
             
-            generate_btn = gr.Button("🚀 이미지 생성", variant="primary", size="lg")
+            generate_btn = gr.Button("🚀 Generate Images", variant="primary", size="lg")
         
         with gr.Column(scale=3):
-            output_gallery = gr.Gallery(label="생성된 이미지", elem_classes="gallery-item", columns=2, object_fit="contain", height=650)
+            output_gallery = gr.Gallery(label="Generated Images", elem_classes="gallery-item", columns=2, object_fit="contain", height=650)
     
-    gr.HTML("""<div class="examples-header"><h3>✨ 예제 프롬프트</h3></div>""")
+    gr.HTML("""<div class="examples-header"><h3>✨ Example Prompts</h3></div>""")
     gr.Examples(
         examples=example_prompts,
         inputs=[prompt, negative_prompt, steps, guidance, width, height, num_samples],
@@ -147,7 +148,7 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
         cache_examples=True,
     )
     
-    # 이벤트 연결
+    # Event connections
     generate_btn.click(
         fn=generate,
         inputs=[prompt, negative_prompt, steps, guidance, width, height, num_samples],
@@ -156,8 +157,8 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
     
     gr.HTML("""
     <div style="text-align: center; margin-top: 20px; padding: 10px; background-color: #f0f0f0; border-radius: 10px;">
-        <p>💡 팁: 고품질 이미지를 위해서는 세부적인 프롬프트와 더 높은 추론 단계를 사용하세요.</p>
-        <p>예) "professional photography, 8k, highly detailed, sharp focus, HDR" 등의 품질 관련 용어를 추가해보세요.</p>
+        <p>💡 Tip: For high-quality images, use detailed prompts and higher inference steps.</p>
+        <p>Example: Add quality terms like "professional photography, 8k, highly detailed, sharp focus, HDR" to your prompts.</p>
     </div>
     """)