changed app to use cpu

app.py

@@ -18,8 +18,11 @@ from losses import *
 from networks_SRGAN import *
 from utils import *
 
-device = 'cuda'
-
+device = 'cpu'
+if device == 'cuda':
+    dtype = torch.cuda.FloatTensor
+else:
+    dtype = torch.FloatTensor
 
 NetG = Generator()
 model_parameters = filter(lambda p: True, NetG.parameters())
@@ -55,8 +58,6 @@ def predict(img):
     img = lr_transforms(img)
     lr_tensor = utils.image2tensor(img, range_norm=False, half=False)
     
-    device = 'cuda'
-    dtype = torch.cuda.FloatTensor
     xLR = lr_tensor.to(device).unsqueeze(0)
     xLR = xLR.type(dtype)
     # pass them through the network