Update app.py

app.py

@@ -4,7 +4,7 @@ import json
 import logging
 import torch
 from PIL import Image
-import spaces
+# 'spaces' 모듈 제거 (필요시 올바른 모듈로 대체)
 from diffusers import (
     DiffusionPipeline,
     AutoencoderTiny,
@@ -73,7 +73,6 @@ pipe_controlnet = FluxControlNetPipeline(
     transformer=pipe.transformer,  # unet 대신 transformer 사용
     controlnet=controlnet,
     scheduler=pipe.scheduler
-    # 'safety_checker'와 'feature_extractor'는 제거됨
 ).to(device)  # 'torch_dtype' 제거
 
 MAX_SEED = 2**32 - 1
@@ -305,7 +304,7 @@ def remove_custom_lora(selected_indices, current_loras):
         lora_image_2
     )
 
-@spaces.GPU(duration=75)
+@spaces.GPU(duration=75)  # 데코레이터 제거가 필요할 수 있음
 def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress):
     print("Generating image...")
     pipe.to(device)
@@ -325,7 +324,7 @@ def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress)
         ):
             yield img
 
-@spaces.GPU(duration=75)
+@spaces.GPU(duration=75)  # 데코레이터 제거가 필요할 수 있음
 def generate_image_to_image(prompt_mash, image_input_path, image_strength, steps, cfg_scale, width, height, seed):
     pipe_i2i.to(device)
     generator = torch.Generator(device=device).manual_seed(seed)
@@ -429,7 +428,7 @@ def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_ind
         
         yield final_image, seed, gr.update(value=progress_bar, visible=False)
 
-run_lora.zerogpu = True
+run_lora.zerogpu = True  # 이 줄도 데코레이터 문제로 제거 필요할 수 있음
 
 def get_huggingface_safetensors(link):
     split_link = link.split("/")
@@ -516,7 +515,7 @@ def process_input(input_image, upscale_factor, **kwargs):
 
     return input_image.resize((w, h)), w_original, h_original, was_resized
 
-@spaces.GPU(duration=75)
+@spaces.GPU(duration=75)  # 데코레이터 제거가 필요할 수 있음
 def infer_upscale(
     seed,
     randomize_seed,
@@ -662,7 +661,7 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, delete_cache=(60, 3600)) as a
         # 이벤트 핸들러 설정
         generate_button.click(
             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],
+            inputs=[prompt, image_input, 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),
@@ -671,7 +670,7 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, delete_cache=(60, 3600)) as a
         )
         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],
+            inputs=[prompt, image_input, 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),
@@ -764,3 +763,4 @@ app.queue()
 app.launch()
 
 
+