quickfix

ui/utils.py

@@ -6,17 +6,22 @@ import torch
 import numpy as np
 from PIL import Image
 import torchvision.transforms as transforms
+import os
 
 IMAGE_SIZE = (256, 256)
 
 NORMALIZE_MEAN = [0.485, 0.456, 0.406]
 NORMALIZE_STD = [0.229, 0.224, 0.225]
 
-CLASS_NAMES_FILE = "classNames.txt" 
 TOP_K_PREDICTIONS = 5
 CONFIDENCE_THRESHOLD = 0.01
 
 
+BASE_DIR = os.path.dirname(os.path.abspath(__file__))
+
+# Path to classNames.txt relative to this file
+CLASS_NAMES_FILE = os.path.join(BASE_DIR, "classNames.txt")
+
 with open(CLASS_NAMES_FILE, "r") as f:
     CLASS_NAMES = [line.strip() for line in f.readlines() if line.strip()]