Spaces:

tonyassi
/

image-blender

Running on Zero

tonyassi commited on May 28

Commit

4386d14

•

1 Parent(s): c6d89ea

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ import spaces
 from diffusers import KandinskyPriorPipeline, KandinskyPipeline
 from diffusers.utils import load_image
 import torch
 pipe_prior = KandinskyPriorPipeline.from_pretrained(
     "kandinsky-community/kandinsky-2-1-prior", torch_dtype=torch.float16

 from diffusers import KandinskyPriorPipeline, KandinskyPipeline
 from diffusers.utils import load_image
 import torch
+from PIL import Image
 pipe_prior = KandinskyPriorPipeline.from_pretrained(
     "kandinsky-community/kandinsky-2-1-prior", torch_dtype=torch.float16