Update app.py

app.py

@@ -1,17 +1,20 @@
 import torch
-import os
+import matplotlib.pyplot as plt
+import pandas as pd
+import seaborn as sns
+from ultralytics import YOLO
 
-# Download the model from Hugging Face
-model_path = "evoSEGA.pt"
-if not os.path.exists(model_path):
-    torch.hub.download_url_to_file(
-        "https://huggingface.co/DrChamyoung/EvoSEGA/resolve/main/evoSEGA.pt",
-        model_path
-    )
+# Path to your model file
+MODEL_PATH = "EvoSEGA/evoSEGA.pt"  # relative path inside repo
 
-# Load YOLOv5 model from .pt file
-model = torch.hub.load('ultralytics/yolov5', 'custom', path=model_path, source='github')
+# Load YOLO model
+model = YOLO(MODEL_PATH)
 
-# Test with an image
-results = model("https://ultralytics.com/images/zidane.jpg")
-results.show()  # Show detection results
+# Example prediction (replace 'test.jpg' with your image)
+results = model("test.jpg")
+
+# Print results
+for r in results:
+    print(r.boxes)
+    # Optionally plot
+    r.show()