Update app.py

app.py

@@ -23,9 +23,11 @@ import warnings
 
 huggingface_token = os.getenv("HUGGINFACE_TOKEN")
 
-# 번역 모델 로드
-translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en", device=0 if torch.cuda.is_available() else -1)
 
+translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en", device="cpu")
+
+
+        
 #Load prompts for randomization
 df = pd.read_csv('prompts.csv', header=None)
 prompt_values = df.values.flatten()
@@ -353,8 +355,11 @@ def generate_image_to_image(prompt_mash, image_input_path, image_strength, steps
     ).images[0]
     return final_image
 
+# run_lora 함수 수정
 def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_indices, lora_scale_1, lora_scale_2, randomize_seed, seed, width, height, loras_state, progress=gr.Progress(track_tqdm=True)):
-    # 한글 감지 및 번역
+    try:
+        
+
     if any('\u3131' <= char <= '\u318E' or '\uAC00' <= char <= '\uD7A3' for char in prompt):
         translated = translator(prompt, max_length=512)[0]['translation_text']
         print(f"Original prompt: {prompt}")
@@ -418,25 +423,28 @@ def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_ind
         if randomize_seed:
             seed = random.randint(0, MAX_SEED)
 
-    # Generate image
-    if image_input is not None:
-        final_image = generate_image_to_image(prompt_mash, image_input, image_strength, steps, cfg_scale, width, height, seed)
-        yield final_image, seed, gr.update(visible=False)
-    else:
-        image_generator = generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress)
-        # Consume the generator to get the final image
-        final_image = None
-        step_counter = 0
-        for image in image_generator:
-            step_counter += 1
-            final_image = image
-            progress_bar = f'<div class="progress-container"><div class="progress-bar" style="--current: {step_counter}; --total: {steps};"></div></div>'
-            yield image, seed, gr.update(value=progress_bar, visible=True)
-        
-        if final_image is None:
-            raise gr.Error("Failed to generate image")
-        
-        yield final_image, seed, gr.update(value=progress_bar, visible=False)
+        if image_input is not None:
+            final_image = generate_image_to_image(prompt_mash, image_input, image_strength, steps, cfg_scale, width, height, seed)
+            return final_image, seed, gr.update(visible=False)
+        else:
+            image_generator = generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress)
+            final_image = None
+            step_counter = 0
+            for image in image_generator:
+                step_counter += 1
+                final_image = image
+                progress_bar = f'<div class="progress-container"><div class="progress-bar" style="--current: {step_counter}; --total: {steps};"></div></div>'
+                yield image, seed, gr.update(value=progress_bar, visible=True)
+            
+            if final_image is None:
+                raise gr.Error("Failed to generate image")
+            
+            return final_image, seed, gr.update(value=progress_bar, visible=False)
+    except Exception as e:
+        print(f"Error in run_lora: {str(e)}")
+        return None, seed, gr.update(visible=False)
+
+
 
 run_lora.zerogpu = True
 
@@ -741,16 +749,18 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, delete_cache=(60, 3600)) as a
         inputs=[selected_indices, loras_state],
         outputs=[loras_state, gallery, selected_info_1, selected_info_2, selected_indices, lora_scale_1, lora_scale_2, lora_image_1, lora_image_2]
     )
+# 이벤트 핸들러 수정
     gr.on(
         triggers=[generate_button.click, prompt.submit],
         fn=run_lora,
         inputs=[prompt, input_image, image_strength, cfg_scale, steps, selected_indices, lora_scale_1, lora_scale_2, randomize_seed, seed, width, height, loras_state],
         outputs=[result, seed, progress_bar]
-    ).then(  # Update the history gallery
-        fn=lambda x, history: update_history(x, history),
+    ).then(
+        fn=lambda x, history: update_history(x, history) if x is not None else history,
         inputs=[result, history_gallery],
         outputs=history_gallery,
     )
+
     
     # 업스케일 버튼 이벤트 추가
     upscale_button.click(