Upload env.py

env.py

@@ -45,6 +45,9 @@ def calculate_iou(box1: List[List[int]], box2: List[List[int]]) -> float:
 class MetaOCTEnv:
     def __init__(self, data_dir: str = "images", truth_file: str = "ground_truth.json", difficulty: str = "medium"):
         self.data_dir = data_dir
+        # Support both flat root upload (HuggingFace) and local images/ folder
+        if not os.path.isdir(data_dir):
+            self.data_dir = "."  # Fallback to root directory
         with open(truth_file, "r") as f:
             self.ground_truth = json.load(f)
         self.image_files = list(self.ground_truth.keys())