Upload 2 files

app.py

@@ -4,7 +4,7 @@ from PIL import Image
 import torch
 from torchvision.transforms import ToTensor
 from torchvision import transforms
-
+from model import pixact
 
 # Load the ONNX model
 model_path = "./SRnet.pth"  # Replace with your model path
@@ -27,7 +27,7 @@ def superresolve(image):
 
 
   # Run inference
-  output = net(image)
+  output = pixact(net(image))
 
   # Postprocess the output
   output = output.permute(0,2,3,1)[0].data.numpy()

model.py

@@ -3,6 +3,11 @@ import torch.nn as nn
 import torch.nn.init as init
 import math
 
+
+def pixact(x):
+    #return (torch.tanh(x) + 1) / 2
+    return x.sigmoid()
+
 # class Net(nn.Module):
 #     def __init__(self, upscale_factor):
 #         super(Net, self).__init__()
@@ -239,7 +244,7 @@ class Net(nn.Module):
         x = self.conv6(x)
         x = self.conv7(x)
         x = self.conv7_1(x)
-        x = self.relu(self.conv8(x))
+        x = self.conv8(x)
         return x
 
     # def _initialize_weights(self):