bug fix

app.py

@@ -19,7 +19,7 @@ vit.load_state_dict(torch.load(f="pretrained_vit_festure_extractor_flower_classi
 def predict(img) -> Tuple[Dict, float]:
   """transforms and perfroms a prediction on img and returns prediction and time taken"""
   #start the time
-  start_time = time.time()
+  start_time = timer()
   #transform the target image and add a batch dim
   img = vit_transforms(img).unsqueeze(0)
   #put the model in eval mode and turn on inference
@@ -30,7 +30,7 @@ def predict(img) -> Tuple[Dict, float]:
   # create a prediction label and prediction probability
   pred_labels_and_probs = {class_names[i]: float(pred_probs[0][i]) for i in range(len(class_names))}
   # calculate prediction time
-  end_time = time.time()
+  end_time = timer()
   pred_time = round(end_time - start_time, 5)
   # return the prediction dictionary and prediction time
   return pred_labels_and_probs, pred_time