Update generationPipeline.py

generationPipeline.py

@@ -3,6 +3,7 @@ import torch.nn as nn
 import numpy as np
 from sampler import DDPMSampler
 from tqdm import tqdm
+from PIL import Image
 
 
 WIDTH = 512
@@ -82,7 +83,8 @@ def generate(
 
         latents_shape = (1, 4, LATENTS_HEIGHT, LATENTS_WIDTH)
 
-        if input_image.any():
+        if input_image:
+            input_image = Image.open(input_image)
             encoder = models["encoder"]
             encoder.to(device)