Update app-backup.py

app-backup.py

@@ -24,6 +24,9 @@ from transformers import T5EncoderModel, T5Tokenizer
 # from optimum.quanto import freeze, qfloat8, quantize
 from transformers import pipeline
 
+ko_translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
+ja_translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ja-en")
+
 class HFEmbedder(nn.Module):
     def __init__(self, version: str, max_length: int, **hf_kwargs):
         super().__init__()
@@ -746,12 +749,8 @@ model = Flux().to(dtype=torch.bfloat16, device="cuda")
 result = model.load_state_dict(sd)
 model_zero_init = False
 
-
-
-ko_translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
-ja_translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ja-en")
-zh_translator = pipeline("translation", model="Helsinki-NLP/opus-mt-zh-en")
-
+# model = Flux().to(dtype=torch.bfloat16, device="cuda")
+# result = model.load_state_dict(load_file("/storage/dev/nyanko/flux-dev/flux1-dev.sft"))
 
 
 @spaces.GPU
@@ -763,17 +762,14 @@ def generate_image(
 ):
     translated_prompt = prompt
     
-    # 한글, 일본어, 중국어 문자 감지
+    # 한글 또는 일본어 문자 감지
     def contains_korean(text):
         return any('\u3131' <= c <= '\u318E' or '\uAC00' <= c <= '\uD7A3' for c in text)
         
     def contains_japanese(text):
         return any('\u3040' <= c <= '\u309F' or '\u30A0' <= c <= '\u30FF' or '\u4E00' <= c <= '\u9FFF' for c in text)
     
-    def contains_chinese(text):
-        return any('\u4e00' <= c <= '\u9fff' for c in text)
-    
-    # 한글, 일본어, 중국어가 있으면 번역
+    # 한글이나 일본어가 있으면 번역
     if contains_korean(prompt):
         translated_prompt = ko_translator(prompt, max_length=512)[0]['translation_text']
         print(f"Translated Korean prompt: {translated_prompt}")
@@ -782,14 +778,6 @@ def generate_image(
         translated_prompt = ja_translator(prompt, max_length=512)[0]['translation_text'] 
         print(f"Translated Japanese prompt: {translated_prompt}")
         prompt = translated_prompt
-    elif contains_chinese(prompt):
-        translated_prompt = zh_translator(prompt, max_length=512)[0]['translation_text']
-        print(f"Translated Chinese prompt: {translated_prompt}")
-        prompt = translated_prompt
-
-
-
-    
 
     if seed == 0:
         seed = int(random.random() * 1000000)
@@ -855,10 +843,10 @@ footer {
 
 def create_demo():
     with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as demo:
-
+    gr.Markdown("# FLUXllama Multilingual")        
         with gr.Row():
             with gr.Column():
-                prompt = gr.Textbox(label="Prompt(한글 가능)", value="A cute and fluffy golden retriever puppy sitting upright, holding a neatly designed white sign with bold, colorful lettering that reads 'Have a Happy Day!' in cheerful fonts. The puppy has expressive, sparkling eyes, a happy smile, and fluffy ears slightly flopped. The background is a vibrant and sunny meadow with soft-focus flowers, glowing sunlight filtering through the trees, and a warm golden glow that enhances the joyful atmosphere. The sign is framed with small decorative flowers, adding a charming and wholesome touch. Ensure the text on the sign is clear and legible.")
+                prompt = gr.Textbox(label="Prompt(Supports English, Korean, and Japanese)", value="A cute and fluffy golden retriever puppy sitting upright, holding a neatly designed white sign with bold, colorful lettering that reads 'Have a Happy Day!' in cheerful fonts. The puppy has expressive, sparkling eyes, a happy smile, and fluffy ears slightly flopped. The background is a vibrant and sunny meadow with soft-focus flowers, glowing sunlight filtering through the trees, and a warm golden glow that enhances the joyful atmosphere. The sign is framed with small decorative flowers, adding a charming and wholesome touch. Ensure the text on the sign is clear and legible.")
                 
                 width = gr.Slider(minimum=128, maximum=2048, step=64, label="Width", value=768)
                 height = gr.Slider(minimum=128, maximum=2048, step=64, label="Height", value=768)
@@ -880,6 +868,17 @@ def create_demo():
             with gr.Column():
                 output_image = gr.Image(label="Generated Image")
                 output_seed = gr.Text(label="Used Seed")
+                output_translated = gr.Text(label="Translated Prompt")
+
+        # Examples 컴포넌트 추가
+        gr.Examples(
+            examples=[
+                "a tiny astronaut hatching from an egg on the moon",
+                "썬글라스 착용한 귀여운 흰색 고양이가 'LOVE'라는 표지판을 들고있다", 
+                "桜が流れる夜の街、照明",
+            ],
+            inputs=prompt,  # 예제가 입력될 컴포넌트 지정
+        )
 
         do_img2img.change(
             fn=lambda x: [gr.update(visible=x), gr.update(visible=x), gr.update(visible=x)],
@@ -890,14 +889,8 @@ def create_demo():
         generate_button.click(
             fn=generate_image,
             inputs=[prompt, width, height, guidance, inference_steps, seed, do_img2img, init_image, image2image_strength, resize_img],
-            outputs=[output_image, output_seed]
+            outputs=[output_image, output_seed, output_translated]
         )
-        
-        examples = [
-            "a tiny astronaut hatching from an egg on the moon",
-            "a cat holding a sign that says hello world",
-            "an anime illustration of a wiener schnitzel",
-        ]
 
     return demo