fpramunno commited on
Commit
8698210
1 Parent(s): 9f24831

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -69,7 +69,7 @@ def generate_image(seed_image):
69
 
70
  generated_image = diffusion.sample(model, y=input_img_pil, labels=None, n=1)
71
 
72
- inp_img = seed_image_tensor.reshape(1, 256, 256).permute(1, 2, 0)
73
  inp_img = np.squeeze(inp_img.cpu().numpy())
74
  inp = Image.fromarray(inp_img) # Create a PIL Image from array
75
  inp = inp.transpose(Image.FLIP_TOP_BOTTOM)
 
69
 
70
  generated_image = diffusion.sample(model, y=input_img_pil, labels=None, n=1)
71
 
72
+ inp_img = input_img_pil.reshape(1, 256, 256).permute(1, 2, 0)
73
  inp_img = np.squeeze(inp_img.cpu().numpy())
74
  inp = Image.fromarray(inp_img) # Create a PIL Image from array
75
  inp = inp.transpose(Image.FLIP_TOP_BOTTOM)