Update app.py

app.py

@@ -13,7 +13,10 @@ accelerator = Accelerator(cpu=True)
 pipe = accelerator.prepare(AmusedPipeline.from_pretrained("amused/amused-512", variant=None, torch_dtype=torch.float32, use_safetensors=True))
 pipe.vqvae.to(torch.float32)
 pipe = pipe.to("cpu")
+apol=[]
 def plex(prompt):
+    gc.collect()
+    apol=[]
     prompt = prompt
     generator = torch.Generator(device="cpu").manual_seed(random.randint(1, 4836923))
     image = pipe(prompt, generator=generator)