Modified Title and test_transforms

app.py

@@ -8,7 +8,7 @@ from custom_torch_module import setup_utils
 
 
 
-title = "Age Prediction model"
+title = "Age-PT"
 description = "ViT(medium clip) based model. transfer trained with custom dataset"
 article = "Through bunch of fine tuning and experiments. REMEMBER! This model can be wrong."
 
@@ -27,10 +27,10 @@ model_classifier = nn.Sequential(nn.Linear(512, 512),
                                 nn.GELU(),
                                 nn.Linear(512, 1))
 model = nn.Sequential(model, model_classifier)
-
-test_transform = setup_utils.build_transform(img_size=224, is_data_aug=False)
 model.load_state_dict(state_dict=torch.load(FILE_NAME, weights_only=True))
 
+test_transform = timm.data.create_transform(input_size=224, is_training=False, interpolation="bicubic")
+