Update app.py

app.py

@@ -4,7 +4,6 @@ import json
 import logging
 import torch
 from PIL import Image
-import spaces
 from diffusers import DiffusionPipeline, AutoencoderTiny, AutoencoderKL, AutoPipelineForImage2Image
 from live_preview_helpers import calculate_shift, retrieve_timesteps, flux_pipe_call_that_returns_an_iterable_of_images
 from diffusers.utils import load_image
@@ -79,7 +78,6 @@ def update_selection(evt: gr.SelectData, width, height):
         height,
     )
 
-@spaces.GPU(duration=70)
 def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, lora_scale, progress):
     pipe.to("cuda")
     generator = torch.Generator(device="cuda").manual_seed(seed)
@@ -116,7 +114,6 @@ def generate_image_to_image(prompt_mash, image_input_path, image_strength, steps
     ).images[0]
     return final_image 
 
-@spaces.GPU(duration=70)
 def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_index, randomize_seed, seed, width, height, lora_scale, progress=gr.Progress(track_tqdm=True)):
     if selected_index is None:
         raise gr.Error("You must select a LoRA before proceeding.")
@@ -155,7 +152,6 @@ def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_ind
             seed = random.randint(0, MAX_SEED)
             
     if(image_input is not None):
-        
         final_image = generate_image_to_image(prompt_mash, image_input, image_strength, steps, cfg_scale, width, height, lora_scale, seed)
         yield final_image, seed, gr.update(visible=False)
     else:
@@ -240,7 +236,6 @@ def get_huggingface_safetensors(link):
 
     return split_link[1], link, safetensors_name, trigger_word, image_url
 
-
 def check_custom_model(link):
     if(link.startswith("https://")):
         if(link.startswith("https://huggingface.co") or link.startswith("https://www.huggingface.co")):
@@ -290,8 +285,6 @@ def add_custom_lora(custom_lora):
 def remove_custom_lora():
     return gr.update(visible=False), gr.update(visible=False), gr.update(), "", None, ""
 
-run_lora.zerogpu = True
-
 css = '''
 #gen_btn{height: 100%}
 #gen_column{align-self: stretch}
@@ -308,8 +301,9 @@ css = '''
 .progress-container {width: 100%;height: 30px;background-color: #f0f0f0;border-radius: 15px;overflow: hidden;margin-bottom: 20px}
 .progress-bar {height: 100%;background-color: #4f46e5;width: calc(var(--current) / var(--total) * 100%);transition: width 0.5s ease-in-out}
 '''
+
 font=[gr.themes.GoogleFont("Source Sans Pro"), "Arial", "sans-serif"]
-with gr.Blocks(theme=gr.themes.Soft(font=font), css=css, delete_cache=(60, 60)) as app:
+with gr.Blocks(theme=gr.themes.Soft(font=font), css=css) as app:
     title = gr.HTML(
         """<h1><img src="https://huggingface.co/spaces/multimodalart/flux-lora-the-explorer/resolve/main/flux_lora.png" alt="LoRA"> FLUX LoRA the Explorer</h1>""",
         elem_id="title",