Fix the transformer error

app.py

@@ -31,24 +31,18 @@ def predict(img) -> Tuple[Dict, float]:
   # Start a timer
   start_time = timer()
   # Transform the input image for use with EffNetB2
-  transform_img = effnetb2_transformer(img).unsqueeze(0)
+  transform_img = effnetb2_transforms(img).unsqueeze(0)
 
   # Put model into eval mode, main prediction
   effnetb2.eval()
   with torch.inference_mode():
-    pred=effnetb2(transform_img)
-    pred_prob = torch.softmax(pred,dim=1)
+    pred_prob=torch.softmax(effnetb2(transform_img),dim=1)
 
-
-
-    
-
-  end_time = timer()
   # Create a prediction label and prediction probability dictionary
   pred_labels_and_probs = {class_names[i]:float(pred_prob[0][i]) for i in range(len(class_names))}
   
   # Calculate pred time
-  time = round(end_time-start_time,4)
+  time = round(timer()-start_time,4)
 
   # Return pred dict and pred time
   return pred_labels_and_probs,time