Update app.py

app.py

@@ -2,25 +2,16 @@ import torch
 import matplotlib.pyplot as plt
 import pandas as pd
 import seaborn as sns
-from ultralytics import YOLO
-import os
 
-# Correct path to your model file
-MODEL_PATH = "models/evoSEGA.pt"  # exact case match to repo
+# Path to model
+MODEL_PATH = "Model/EvoSEGA/evoSEGA.pt"
 
-if not os.path.exists(MODEL_PATH):
-    raise FileNotFoundError(f"Model file not found at {MODEL_PATH}")
+# Load YOLOv5 model
+model = torch.hub.load("ultralytics/yolov5", "custom", path=MODEL_PATH, force_reload=True)
 
-# Load YOLO model
-model = YOLO(MODEL_PATH)
+# Run inference
+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)
-
-for r in results:
-    print(r.boxes)
-    r.show()
+# Print results
+print(results.pandas().xyxy[0])  # Pandas dataframe
+results.show()  # Show image with detections