Upload folder using huggingface_hub

app.py

@@ -17,9 +17,7 @@ else: device = 'cpu'
 print('DEVICE:', device)
 if device == 'cuda': print('CUDA DEVICE:', torch.cuda.get_device_name())
 
-torch.backends.cudnn.benchmark = True
-
-model = VGG_19().to(device)
+model = VGG_19().to(device).eval()
 for param in model.parameters():
     param.requires_grad = False