Update app.py

app.py

@@ -33,7 +33,7 @@ def process_img(image):
     y = torch.tensor(img).permute(2,0,1).unsqueeze(0).to(device)
 
     with torch.no_grad():
-        x_hat = model(y)
+        x_hat, _, _, _ = model(y)
 
     restored_img = x_hat.squeeze().permute(1,2,0).clamp_(0, 1).cpu().detach().numpy()
     restored_img = np.clip(restored_img, 0. , 1.)