Update app.py

app.py

@@ -1,14 +1,16 @@
 from PIL import Image
 import gradio as gr
 import numpy as np
+import PIL.Image
 import random
+import cv2
 import torch
 from accelerate import Accelerator
 from transformers import pipeline
 from diffusers.utils import load_image
 from diffusers import KandinskyV22PriorPipeline, KandinskyV22ControlnetPipeline
 
-accelerator = Accelerator()
+accelerator = Accelerator(cpu=True)
 
 def make_hint(image, depth_estimator):
     image = depth_estimator(image)["depth"]
@@ -24,6 +26,7 @@ pipe_prior = accelerator.prepare(KandinskyV22PriorPipeline.from_pretrained("kand
 pipe_prior = accelerator.prepare(pipe_prior.to("cpu"))
 pipe = accelerator.prepare(KandinskyV22ControlnetPipeline.from_pretrained("kandinsky-community/kandinsky-2-2-controlnet-depth", torch_dtype=torch.float32))
 pipe = accelerator.prepare(pipe.to("cpu"))
+pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
 generator = torch.Generator(device="cpu").manual_seed(random.randint(-1, 2147483647))
 
 def plex(prompt,goof):