Update app.py

app.py

@@ -104,7 +104,7 @@ def generate_image(orginal_image, prompt, adapter_names,  steps, seed, image_str
                 generator=generator,
                 joint_attention_kwargs=joint_attention_kwargs
             ).images[0]
-    
+    torch.cuda.empty_cache() 
     progress(99, "Generate image success!")
     return generated_image
 
@@ -150,7 +150,7 @@ def upload_image_to_r2(image, account_id, access_key, secret_key, bucket_name):
 def run_lora(prompt, image_url, lora_strings_json, image_strength,  cfg_scale, steps, randomize_seed, seed, width, height, upload_to_r2, account_id, access_key, secret_key, bucket, progress=gr.Progress(track_tqdm=True)):
     print("run_lora", prompt, lora_strings_json, cfg_scale, steps, width, height)
     gr.Info("Starting process")
-
+    torch.cuda.empty_cache() 
     img2img_model = False
     orginal_image = None
     if image_url: