Update app.py
Browse files
app.py
CHANGED
@@ -31,7 +31,7 @@ def generate_image(seed_image):
|
|
31 |
seed_image_tensor = transform_hmi(Image.open(seed_image)).reshape(1, 1, 256, 256).to(device)
|
32 |
generated_image = diffusion.sample(model, y=seed_image_tensor, labels=None, n=1)
|
33 |
# generated_image_pil = transforms.ToPILImage()(generated_image.squeeze().cpu())
|
34 |
-
img = generated_image[0].permute(1, 2, 0) # Permute dimensions to height x width x channels
|
35 |
img = np.squeeze(img.cpu().numpy())
|
36 |
v = Image.fromarray(img) # Create a PIL Image from array
|
37 |
v = v.transpose(Image.FLIP_TOP_BOTTOM)
|
|
|
31 |
seed_image_tensor = transform_hmi(Image.open(seed_image)).reshape(1, 1, 256, 256).to(device)
|
32 |
generated_image = diffusion.sample(model, y=seed_image_tensor, labels=None, n=1)
|
33 |
# generated_image_pil = transforms.ToPILImage()(generated_image.squeeze().cpu())
|
34 |
+
img = generated_image[0].reshape(1, 256, 256).permute(1, 2, 0) # Permute dimensions to height x width x channels
|
35 |
img = np.squeeze(img.cpu().numpy())
|
36 |
v = Image.fromarray(img) # Create a PIL Image from array
|
37 |
v = v.transpose(Image.FLIP_TOP_BOTTOM)
|