Update app.py

app.py

@@ -109,6 +109,11 @@ def load_model(device, repo_id, filename):
 
 def preprocess_image(image):
     try:
+        # Ensure the image is a PIL Image
+        if isinstance(image, np.ndarray):
+            image = Image.fromarray(image) # Convert NumPy array to PIL Image
+
+        # Resize and convert to tensor
         transform = transforms.Compose([
             transforms.Resize((image_size, image_size)),
             transforms.ToTensor(),