Update app.py

app.py

@@ -57,7 +57,7 @@ pipe_i2i = AutoPipelineForImage2Image.from_pretrained(
     text_encoder_2=pipe.text_encoder_2,
     tokenizer_2=pipe.tokenizer_2,
     torch_dtype=dtype
-)
+).to(device)
 
 # 업스케일링을 위한 ControlNet 모델 로드
 controlnet = FluxControlNetModel.from_pretrained(
@@ -74,10 +74,7 @@ pipe_controlnet = FluxControlNetPipeline(
     controlnet=controlnet,
     scheduler=pipe.scheduler
     # 'safety_checker'와 'feature_extractor'는 제거됨
-).to(device)
-
-
-
+).to(device)  # 'torch_dtype' 제거
 
 MAX_SEED = 2**32 - 1
 MAX_PIXEL_BUDGET = 1024 * 1024
@@ -667,11 +664,19 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, delete_cache=(60, 3600)) as a
             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),
+            inputs=[result, history_gallery],
+            outputs=history_gallery,
         )
         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),
+            inputs=[result, history_gallery],
+            outputs=history_gallery,
         )
         gallery.select(
             fn=update_selection,
@@ -703,11 +708,6 @@ 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]
         )
-        history_gallery.load(
-            fn=update_history,
-            inputs=[result, history_gallery],
-            outputs=history_gallery
-        )
 
     with gr.Tab("Upscale"):
         with gr.Row():
@@ -763,3 +763,4 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, delete_cache=(60, 3600)) as a
 app.queue()
 app.launch()
 
+