ZeroGPU

app.py

@@ -9,7 +9,7 @@ import os
 import imquality.brisque as brisque
 from loss.niqe_utils import *
 
-eval_net = CIDNet()
+eval_net = CIDNet().cuda()
 eval_net.trans.gated = True
 eval_net.trans.gated2 = True
 
@@ -31,8 +31,8 @@ def process_image(input_img,score,model_path,gamma=1.0,alpha_s=1.0,alpha_i=1.0):
     with torch.no_grad():
         eval_net.trans.alpha_s = alpha_s
         eval_net.trans.alpha = alpha_i
-        output = eval_net(input**gamma)
-    output = torch.clamp(output,0,1)
+        output = eval_net(input.cuda()**gamma)
+    output = torch.clamp(output,0,1).cuda()
     output = output[:, :, :h, :w]
     enhanced_img = transforms.ToPILImage()(output.squeeze(0))
     if score == 'Yes':