modify classify image fn

app.py

@@ -9,10 +9,10 @@ from models.convmodel import MNISTnet
 from pathlib import Path 
 
 # Function to perform image classification
-def classify_image(image_path):
-    imdata = np.asarray(Image.open(image_path))
+def classify_image(img):
+    #imdata = np.asarray(Image.open(image_path))
     alltransforms = torchvision.transforms.Compose([torchvision.transforms.ToTensor()])
-    tensor_image = alltransforms(imdata)
+    tensor_image = alltransforms(img)
     # bring it to the shape model expects N, C, H, W
     #print(tensor_image.shape)
     model_input_tensor_image = tensor_image.unsqueeze(dim=0)