bug fix

app.py

@@ -10,7 +10,7 @@ from typing import Tuple, Dict
 class_names = ['daisy', 'dandelion', 'rose', 'sunflower', 'tulip']
 ### model and transforms preparation
 #create vit model
-vit, vit_transforms = create_vit_model(num_classes=5):
+vit, vit_transforms = create_vit_model(num_classes=5)
 #load saved weights
 vit.load_state_dict(torch.load(f="pretrained_vit_festure_extractor_flower_classification.pth", 
                                map_location=torch.device('cpu')))