Update app.py

app.py

@@ -9,6 +9,7 @@ from torchvision import datasets, transforms
 from custom_resnet import CustomResNet
 import random
 
+
 model = CustomResNet()
 model.load_state_dict(torch.load('CustomResNet.pth', map_location=torch.device('cpu')), strict=False)
 model.eval()