Update app.py

Add gc to do tensor cleanup

app.py

@@ -6,10 +6,11 @@ from flax.jax_utils import replicate
 from flax.training.common_utils import shard
 from PIL import Image
 from diffusers import FlaxStableDiffusionControlNetPipeline, FlaxControlNetModel
-import cv2
+#import cv2
+import gc
 
-with open("test.html") as f:
-    lines = f.readlines()
+#with open("test.html") as f:
+ #   lines = f.readlines()
 
 def create_key(seed=0):
     return jax.random.PRNGKey(seed)
@@ -63,6 +64,9 @@ def infer(prompts, negative_prompts, image):
     ).images[0]
     
     #output_images = pipe.numpy_to_pil(np.asarray(output.reshape((num_samples,) + output.shape[-3:])))
+    del image
+    gc.collect()
+    
     return output
 
 with gr.Blocks(theme='kfahn/AnimalPose') as demo: