Update pest/source/src/utils.py

pest/source/src/utils.py

@@ -144,9 +144,32 @@ def preprocess_image(image_path: str):
 def postprocess_results(detection_results):
     """Postprocess pest detection results."""
     try:
-        # 加载类别标签
+        import os
+        
+        # 加载类别标签 - 尝试多种路径以适配Hugging Face环境
         class_labels = {}
-        with open('IP102_classes.txt', 'r', encoding='utf-8') as f:
+        possible_class_paths = [
+            'IP102_classes.txt',
+            '/workspace/IP102_classes.txt',
+            '/app/IP102_classes.txt',
+            'src/IP102_classes.txt',
+            'source/src/IP102_classes.txt',
+            'pest/source/src/IP102_classes.txt',
+            '/app/pest/source/src/IP102_classes.txt',
+            '/workspace/pest/source/src/IP102_classes.txt',
+            os.path.join(os.path.dirname(os.path.abspath(__file__)), 'IP102_classes.txt')
+        ]
+        
+        class_file_path = None
+        for path in possible_class_paths:
+            if os.path.exists(path):
+                class_file_path = path
+                break
+        
+        if class_file_path is None:
+            raise FileNotFoundError(f"Class labels file not found: IP102_classes.txt")
+        
+        with open(class_file_path, 'r', encoding='utf-8') as f:
             for line in f:
                 if line.strip():
                     parts = line.strip().split(' ', 1)