Upload app.py

app.py

@@ -19,13 +19,13 @@ def _inner(feat_net, hooks, x):
 
 def _get_layers(arch:str, pretrained=True):
   "Get the layers and arch for a VGG Model (16 and 19 are supported only)"
-  feat_net = vgg19(pretrained=pretrained).cuda() if arch.find('9') > 1 else vgg16(pretrained=pretrained).cuda()
+  feat_net = vgg19(pretrained=pretrained) if arch.find('9') > 1 else vgg16(pretrained=pretrained)
   config = _vgg_config.get(arch)
   features = feat_net.features.cuda().eval()
   for p in features.parameters(): p.requires_grad=False
   return feat_net, [features[i] for i in config]
 
-  
+
 _vgg_config = {
     'vgg16' : [1, 11, 18, 25, 20],
     'vgg19' : [1, 6, 11, 20, 29, 22]