fix: predict_fn bug fix

app.py

@@ -19,7 +19,7 @@ transform = transforms.Compose([
         ])
 
 
-def predict_fn(image_path, raw_output=False):
+def predict_fn(image, raw_output=False):
         """
         This function will predict the class of an image
         :param image_path: The path of the image