Removing CUDA

app.py

@@ -60,13 +60,13 @@ class Wrapper(torch.nn.Module):
   
 def get_layer4(input_image):
   l4_model = models.resnet50(pretrained=True)
-  l4_model = l4_model.cuda()
+  # l4_model = l4_model.cuda()
   l4_model.eval();
   wrapped_model = Wrapper(l4_model)
 
   with torch.no_grad():
     data = imagenet_transform(input_image).unsqueeze(0)
-    data = data.cuda()
+    # data = data.cuda()
     reference_layer4 = wrapped_model(data)
 
   return reference_layer4.data.to('cpu').numpy()