test

app.py

@@ -15,7 +15,7 @@ def pil_to_torch(image):
 def torch_to_pil(image):
     image = image.squeeze(0).cpu().detach().numpy()
     image = image.transpose((1, 2, 0))
-    image = (image * 255).astype(np.uint8)
+    image = (torch.clip(image, 0, 1) * 255).astype(np.uint8)
     image = PIL.Image.fromarray(image)
     return image