Update app.py

app.py

@@ -788,14 +788,13 @@ def get_translator(lang):
             translator = pipeline(
                 "translation",
                 model=model_name,
-                device="cpu",
-                framework="pt"
+                device="cpu"  # CPU로 고정
             )
             translators_cache[lang] = translator
             print(f"Successfully loaded translator for {lang}")
         except Exception as e:
             print(f"Error loading translator for {lang}: {e}")
-            translators_cache[lang] = None
+            return None
             
     return translators_cache[lang]
 
@@ -819,7 +818,6 @@ def translate_prompt(prompt, source_lang):
         print(f"Translation error for {source_lang}: {e}")
         return prompt
 
-
         
 def get_translator(lang):
     if lang == "English":
@@ -863,6 +861,7 @@ def translate_text(text, translator_info):
         print(f"Translation error: {e}")
         return text
 
+
 @spaces.GPU
 @torch.no_grad()
 def generate_image(
@@ -880,7 +879,6 @@ def generate_image(
     except Exception as e:
         print(f"Translation failed: {e}")
         translated_prompt = prompt
-        
 
         
     if seed == 0:
@@ -941,9 +939,6 @@ footer {
 
 def create_demo():
     with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as demo:
-        gr.Markdown("# Multilingual FLUXllama")
-        gr.Markdown("### Supported languages: " + ", ".join(["English"] + sorted(list(TRANSLATORS.keys()))))
-        
         with gr.Row():
             with gr.Column():
                 source_lang = gr.Dropdown(
@@ -956,9 +951,7 @@ def create_demo():
                     label="Prompt",
                     value="A beautiful landscape"
                 )
-                
-
-                
+ 
                 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)
                 guidance = gr.Slider(minimum=1.0, maximum=5.0, step=0.1, label="Guidance", value=3.5)