Update app.py

app.py

@@ -69,10 +69,12 @@ def generate_image(seed_image):
             
     generated_image = diffusion.sample(model, y=input_img_pil, labels=None, n=1)
 
-    inp_img = input_img_pil.reshape(1, 256, 256).permute(1, 2, 0)  
+    inp_img = (inp_img.clamp(-1, 1) + 1) / 2 # to be in [-1, 1], the plus 1 and the division by 2 is to bring back values to [0, 1]
+    inp_img = (inp_img * 255).type(torch.uint8) # to bring in valid pixel range
     inp_img = np.squeeze(inp_img.cpu().numpy())
     inp = Image.fromarray(inp_img)                # Create a PIL Image from array
     inp = inp.transpose(Image.FLIP_TOP_BOTTOM) 
+    
     img = generated_image[0].reshape(1, 256, 256).permute(1, 2, 0)              # Permute dimensions to height x width x channels
     img = np.squeeze(img.cpu().numpy())
     v = Image.fromarray(img)                # Create a PIL Image from array