Update app.py

app.py

@@ -25,6 +25,10 @@ from torchvision.models import resnet18
 
 #device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
 with_points = True
+dim = 128
+dim0 =224
+crop_ratio = dim/dim0
+
 file_savingfolder = './modelbest/'
 ext = '_bestVal'
 core_model_tst = resnet18(pretrained=True)