Update app.py

app.py

@@ -115,27 +115,28 @@ def make_predictions(image_paths):
         # temp = pathlib.PosixPath  # Save the original state
         # pathlib.PosixPath = pathlib.WindowsPath  # Change to WindowsPath temporarily
         
-        model_path = Path(r'model/export')
+        model_path = Path('model/export')
+        if not model_path.exists():
+            return ["Error: Model not found at 'model/export'"]
+
         learner = load_learner(model_path)
-        
         predictions = []
 
-        for image_path in image_paths:
-            # Open the image using fastai's open_image function
-            image = open_image(image_path)
-
-            # Make a prediction
-            prediction_class, prediction_idx, probabilities = learner.predict(image)
-
-            # If you want the predicted class as a string
-            predicted_class_str = str(prediction_class)
-            
-            predictions.append(predicted_class_str)
-
+        for img_path in image_paths:
+            path = Path(img_path)
+            if not path.exists():
+                predictions.append("Error: File not found")
+                continue
+            try:
+                img = open_image(path)
+                pred_class, _, _ = learner.predict(img)
+                predictions.append(str(pred_class))
+            except Exception as e:
+                predictions.append(f"Error: {e}")
         return predictions
 
     except Exception as e:
-        return {"error in make_predictions": str(e)}
+        return [f"Model failed: {e}"]
         
     # finally:
     #     pathlib.PosixPath = temp