Update app.py

app.py

@@ -27,12 +27,11 @@ def truncate_prompt(prompt, max_length=77):
     truncated_prompt = tokenizer.decode(input_ids, skip_special_tokens=True)
     return truncated_prompt
 
-@spaces.GPU()
 def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=4, init_image=None, img2img_strength=0.75, progress=gr.Progress(track_tqdm=True)):
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     generator = torch.Generator(device=device).manual_seed(seed)
-    
+
     # Truncar el prompt si es demasiado largo
     prompt = truncate_prompt(prompt)
     
@@ -40,8 +39,10 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_in
     if init_image and not isinstance(init_image, Image.Image):
         init_image = Image.fromarray(np.array(init_image))
 
+    # Liberar memoria CUDA antes de la generación
+    torch.cuda.empty_cache()
+    
     try:
-        # Generación de la imagen
         if init_image:
             init_image = init_image.convert("RGB")
             generated_image = pipe_img2img(