Update app.py

app.py

@@ -3,6 +3,7 @@ import gradio as gr
 from diffusers import StableDiffusionControlNetPipeline, ControlNetModel, UniPCMultistepScheduler
 import torch
 torch.backends.cuda.matmul.allow_tf32 = True
+import gc
 
 controlnet = ControlNetModel.from_pretrained("ioclab/control_v1p_sd15_brightness", torch_dtype=torch.float16, use_safetensors=True)
 
@@ -29,8 +30,8 @@ def infer(
         seed=1234,
 ):
 
-    conditioning_image = Image.fromarray(conditioning_image)
-    conditioning_image = conditioning_image.convert('L')
+    conditioning_image_raw = Image.fromarray(conditioning_image)
+    conditioning_image = conditioning_image_raw.convert('L')
 
     g_cpu = torch.Generator()
 
@@ -51,6 +52,9 @@ def infer(
         controlnet_conditioning_scale=1.0,
     ).images[0]
 
+    del conditioning_image, conditioning_image_raw
+    gc.collect()
+
     return output_image
 
 with gr.Blocks() as demo: