Update app.py

app.py

@@ -1,5 +1,4 @@
 import os
-import spaces
 import gradio as gr
 import json
 import logging
@@ -304,7 +303,6 @@ def remove_custom_lora(selected_indices, current_loras):
         lora_image_2
     )
 
-@spaces.GPU(duration=75)  # 데코레이터 제거가 필요할 수 있음
 def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress):
     print("Generating image...")
     pipe.to(device)
@@ -324,7 +322,6 @@ def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress)
         ):
             yield img
 
-@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)
@@ -428,7 +425,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("/")
@@ -497,7 +494,8 @@ def process_input(input_image, upscale_factor, **kwargs):
         warnings.warn(
             f"Requested output image is too large ({w * upscale_factor}x{h * upscale_factor}). Resizing to ({int(aspect_ratio * MAX_PIXEL_BUDGET ** 0.5 // upscale_factor), int(MAX_PIXEL_BUDGET ** 0.5 // aspect_ratio // upscale_factor)}) pixels."
         )
-        gr.Info(
+        # Gradio does not have gr.Info, using print instead
+        print(
             f"Requested output image is too large ({w * upscale_factor}x{h * upscale_factor}). Resizing input to ({int(aspect_ratio * MAX_PIXEL_BUDGET ** 0.5 // upscale_factor), int(MAX_PIXEL_BUDGET ** 0.5 // aspect_ratio // upscale_factor)}) pixels budget."
         )
         input_image = input_image.resize(
@@ -515,7 +513,6 @@ def process_input(input_image, upscale_factor, **kwargs):
 
     return input_image.resize((w, h)), w_original, h_original, was_resized
 
-@spaces.GPU(duration=75)  # 데코레이터 제거가 필요할 수 있음
 def infer_upscale(
     seed,
     randomize_seed,
@@ -538,7 +535,8 @@ def infer_upscale(
 
     generator = torch.Generator().manual_seed(seed)
 
-    gr.Info("Upscaling image...")
+    # Gradio does not have gr.Info, using print instead
+    print("Upscaling image...")
     image = pipe_controlnet(
         prompt="",
         control_image=control_image,
@@ -551,7 +549,7 @@ def infer_upscale(
     ).images[0]
 
     if was_resized:
-        gr.Info(
+        print(
             f"Resizing output image to targeted {w_original * upscale_factor}x{h_original * upscale_factor} size."
         )
 
@@ -589,7 +587,7 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, delete_cache=(60, 3600)) as a
 
     loras_state = gr.State(loras)
     selected_indices = gr.State([])
-    
+
     with gr.Tab("Generate"):
         with gr.Row():
             with gr.Column(scale=3):
@@ -609,16 +607,16 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, delete_cache=(60, 3600)) as a
                         lora_scale_1 = gr.Slider(label="LoRA 1 Scale", minimum=0, maximum=3, step=0.01, value=1.15)
                 with gr.Row():
                     remove_button_1 = gr.Button("Remove", size="sm")
-            with gr.Column(scale=8):
-                with gr.Row():
-                    with gr.Column(scale=0, min_width=50):
-                        lora_image_2 = gr.Image(label="LoRA 2 Image", interactive=False, min_width=50, width=50, show_label=False, show_share_button=False, show_download_button=False, show_fullscreen_button=False, height=50)
-                    with gr.Column(scale=3, min_width=100):
-                        selected_info_2 = gr.Markdown("Select a LoRA 2")
-                    with gr.Column(scale=5, min_width=50):
-                        lora_scale_2 = gr.Slider(label="LoRA 2 Scale", minimum=0, maximum=3, step=0.01, value=1.15)
-                with gr.Row():
-                    remove_button_2 = gr.Button("Remove", size="sm")
+        with gr.Column(scale=8):
+            with gr.Row():
+                with gr.Column(scale=0, min_width=50):
+                    lora_image_2 = gr.Image(label="LoRA 2 Image", interactive=False, min_width=50, width=50, show_label=False, show_share_button=False, show_download_button=False, show_fullscreen_button=False, height=50)
+                with gr.Column(scale=3, min_width=100):
+                    selected_info_2 = gr.Markdown("Select a LoRA 2")
+                with gr.Column(scale=5, min_width=50):
+                    lora_scale_2 = gr.Slider(label="LoRA 2 Scale", minimum=0, maximum=3, step=0.01, value=1.15)
+            with gr.Row():
+                remove_button_2 = gr.Button("Remove", size="sm")
         with gr.Row():
             with gr.Column():
                 with gr.Group():
@@ -764,3 +762,4 @@ app.launch()
 
 
 
+