Update app.py

app.py

@@ -11,8 +11,8 @@ import numpy as np
 from accelerate import Accelerator
 accelerator = Accelerator(cpu=True)
 pipe = accelerator.prepare(AmusedPipeline.from_pretrained("amused/amused-512", variant=None, torch_dtype=torch.float32, use_safetensors=True))
-pipe = accelerator.prepare(pipe.vqvae.to(torch.float32))
-pipe = pipe.to("cpu")
+pipe.vqvae.to(torch.float32)
+pipe.to("cpu")
 apol=[]
 def plex(prompt, fifth, twice):
     gc.collect()
@@ -22,7 +22,7 @@ def plex(prompt, fifth, twice):
     while nm % 32 != 0:
         nm = random.randint(1, 4836928)
     generator = torch.Generator(device="cpu").manual_seed(nm)
-    image = pipe([prompt]*fifth, generator=generator,num_inference_steps=twice)
+    image = pipe([prompt]*fifth,num_inference_steps=twice)
     for a, imze in enumerate(image["images"]):
         apol.append(imze)
     return apol