Update app.py

app.py

@@ -3,15 +3,24 @@ import matplotlib.pyplot as plt
 import pandas as pd
 import seaborn as sns
 from ultralytics import YOLO
+import os
 
 # Path to your model file
-MODEL_PATH = "models/evoSEGA.pt"  # relative path inside repo
+MODEL_PATH = "Model/EvoSEGA/evoSEGA.pt"  # Update path to match your repo
+
+# Check if model exists
+if not os.path.exists(MODEL_PATH):
+    raise FileNotFoundError(f"Model file not found at {MODEL_PATH}")
 
 # Load YOLO model
 model = YOLO(MODEL_PATH)
 
-# Example prediction (replace 'test.jpg' with your image)
-results = model("test.jpg")
+# Example prediction
+TEST_IMAGE = "test.jpg"
+if not os.path.exists(TEST_IMAGE):
+    raise FileNotFoundError(f"Test image not found at {TEST_IMAGE}")
+
+results = model(TEST_IMAGE)
 
 # Print results
 for r in results: