fix

app.py

@@ -5,6 +5,7 @@ import torch.nn as nn
 from torchvision import transforms
 import torchvision.models as models
 from PIL import Image
+import numpy as np
 
 # Function to load class names from a file
 def load_class_names(file_path):
@@ -38,7 +39,10 @@ val_transform = transforms.Compose([
 
 # Define the prediction function
 def predict_image(image):
-    image = Image.open(image).convert('RGB')
+    # Convert the NumPy array to a PIL Image
+    if isinstance(image, np.ndarray):
+        image = Image.fromarray(image.astype('uint8'), 'RGB')
+    
     image = val_transform(image)
     image = image.unsqueeze(0)  # Add batch dimension