Update app.py

app.py

@@ -67,19 +67,24 @@ def generate(prompt, negative_prompt, guidance, strength, seed, width, height):
 
     generator = torch.Generator(device="cuda").manual_seed(seed)
 
-    result = pipe(
-        prompt=prompt,
-        negative_prompt=negative_prompt,
-        width=width,
-        height=height,
-        num_inference_steps=NUM_INFERENCE_STEPS,
-        guidance_scale=guidance,
-        generator=generator,
-        callback_on_step_end=callback,
-        callback_on_step_end_tensor_inputs=["latents"],
-    )
-
-    return result.images[0]
+    try:
+        result = pipe(
+            prompt=prompt,
+            negative_prompt=negative_prompt,
+            width=width,
+            height=height,
+            num_inference_steps=NUM_INFERENCE_STEPS,
+            guidance_scale=guidance,
+            generator=generator,
+            callback_on_step_end=callback,
+            callback_on_step_end_tensor_inputs=["latents"],
+        )
+        image = result.images[0]
+        del result
+        return image
+    finally:
+        pipe.disable_lora()
+        torch.cuda.empty_cache()
 
 
 with gr.Blocks() as interface: