Update app.py

app.py

@@ -13,7 +13,6 @@ import numpy as np
 from huggingface_hub import InferenceClient
 import os
 
-
 device = "cuda"
 ckpt_dir = snapshot_download(repo_id="Kwai-Kolors/Kolors")
 ckpt_IPA_dir = snapshot_download(repo_id="Kwai-Kolors/Kolors-IP-Adapter-Plus")
@@ -56,24 +55,6 @@ pipe_i2i.load_ip_adapter(f'{ckpt_IPA_dir}' , subfolder="", weight_name=["ip_adap
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
 
-def call_api(content, system_message, max_tokens=1000, temperature=0.7, top_p=0.95):
-    messages = [{"role": "system", "content": system_message}, {"role": "user", "content": content}]
-    response = client.chat_completion(messages, max_tokens=max_tokens, temperature=temperature, top_p=top_p)
-    return response.choices[0].message['content']
-
-def generate_prompt(korean_prompt):
-    system_message = """
-    Given the following description in Korean, 
-    translate and generate a concise English prompt suitable for a Stable Diffusion model. 
-    The prompt should be focused, descriptive, 
-    and contain specific keywords or phrases that will help guide the image generation process. 
-    Use simple and descriptive language, avoiding unnecessary words. 
-    Ensure the output is in English and follows the format typically used in Stable Diffusion prompts. 
-    The description is: [Insert Korean description here]
-    """
-    optimized_prompt = call_api(korean_prompt, system_message)
-    return optimized_prompt  # 최적화된 프롬프트 반환
-
 @spaces.GPU
 def infer(prompt, 
           ip_adapter_image = None, 
@@ -136,21 +117,6 @@ css="""
 with gr.Blocks(css=css) as Kolors:
     with gr.Row():
         with gr.Column(elem_id="col-left"):
-            with gr.Row():
-                korean_prompt = gr.Textbox(
-                    label="한국어 프롬프트 입력",
-                    placeholder="한국어로 원하는 프롬프트를 입력하세요",
-                    lines=2
-                )
-            with gr.Row():
-                generate_prompt_button = gr.Button("Generate Prompt")
-            with gr.Row():
-                optimized_prompt = gr.Textbox(
-                    label="최적화된 프롬프트 생성",
-                    placeholder=" ",
-                    lines=2,
-                    interactive=False
-                )
             with gr.Row():
                 generated_prompt = gr.Textbox(
                     label="프롬프트 입력",
@@ -219,13 +185,6 @@ with gr.Blocks(css=css) as Kolors:
             result = gr.Image(label="Result", show_label=False)
             download_button = gr.File(label="Download Image")
 
-    # 최적화된 프롬프트 생성 및 결과 표시
-    generate_prompt_button.click(
-        fn=generate_prompt,
-        inputs=[korean_prompt],
-        outputs=[optimized_prompt]
-    )
-    
     # 이미지 생성 및 다운로드 파일 경로 설정
     run_button.click(
         fn=infer,