Minor change

models/SRGAN/srgan.py

@@ -2,7 +2,7 @@ import os
 import torch
 from torch import nn
 from PIL import Image
-from torchvision.transforms import ToTensor
+from torchvision.transforms import ToTensor, functional as TF
 
 class ResidualBlock(nn.Module):
     def __init__(self, in_features):
@@ -62,7 +62,7 @@ class GeneratorResnet(nn.Module):
             x = ToTensor()(x).unsqueeze(0)
             x = self.forward(x)
             x = Image.fromarray((x.squeeze(0).permute(1, 2, 0).detach().numpy() * 255).astype('uint8'))
-        return x
+        return TF.adjust_brightness(x, 1.1)
     
 if __name__ == '__main__':
     current_dir = os.path.dirname(os.path.realpath(__file__))