Made change to models path to take them dynamically

routes/chess_review.py

@@ -6,6 +6,7 @@ from typing import List, Dict
 import asyncio
 import json
 import sys
+import os
 
 if sys.platform == "win32":
     asyncio.set_event_loop_policy(asyncio.WindowsProactorEventLoopPolicy())
@@ -37,8 +38,8 @@ def load_opening_book(csv_path):
     return opening_book
 
 
-engine_path = r"D:\venv\chess-vision\code\stockfish-windows-x86-64-avx2\stockfish\stockfish-windows-x86-64-avx2.exe"
-csv_path = r"D:\venv\chess-vision\code\openings_master.csv"
+engine_path = os.path.join(os.getcwd(), "models", "stockfish_14_x64_avx2.exe")
+csv_path = os.path.join(os.getcwd(), "assets", "opening_book.csv")
 opening_book = load_opening_book(csv_path)
 
 

routes/detection.py

@@ -1,6 +1,10 @@
 from ultralytics import YOLO
 from PIL import Image
-detect_model = YOLO(r'D:\venv\chess-vision\models\chessDetection3d.pt')
+import os 
+
+curr = os.getcwd()
+detect_model_path = os.path.join(curr, 'models', 'chessDetection3d.pt')
+detect_model = YOLO(detect_model_path)
 
 async def detect_pieces(image : Image):
     if image is None:

routes/segmentation.py

@@ -1,7 +1,11 @@
 from ultralytics import YOLO
 from PIL import Image
+import os
 
-seg_model = YOLO(r'D:\venv\chess-vision\models\SegModel (1).pt')
+curr = os.getcwd()
+seg_model_path = os.path.join(curr, 'models', 'SegModel (1).pt')
+
+seg_model = YOLO(seg_model_path)
 
 async def segment_chess_board(image : Image):
     if image is None: