Update app.py

app.py

@@ -18,6 +18,7 @@ class SimpleRealUpscaler:
         model.eval()
         self.model = model.to(self.device)
 
+    @torch.no_grad()
     def upscale(self, np_image_rgb):
         image_rgb_tensor = torch.tensor(np_image_rgb[:,:,::-1].astype(np.float32)).to(self.device)
         image_rgb_tensor /= 255