Update app.py

app.py

@@ -9,11 +9,11 @@ from diffusers.utils import load_image
 from accelerate import Accelerator
 
 accelerator = Accelerator()
-
+apol=[]
+pipe = accelerator.prepare(AutoPipelineForText2Image.from_pretrained("openskyml/overall-v1", torch_dtype=torch.float32, variant=None, use_safetensors=False, safety_checker=None))
+pipe = accelerator.prepare(pipe.to("cpu"))
 def plex(prompt,neg_prompt):
-    apol=[]
-    pipe = accelerator.prepare(AutoPipelineForText2Image.from_pretrained("openskyml/overall-v1", torch_dtype=torch.float32, variant=None, use_safetensors=False, safety_checker=None))
-    pipe = accelerator.prepare(pipe.to("cpu"))
+    generator = torch.Generator(device="cpu").manual_seed(random.randint(1, 4836923))
     image = pipe(prompt=prompt, negative_prompt=neg_prompt, generator=generator, num_inference_steps=15)
     for a, imze in enumerate(image["images"]):
         apol.append(imze)