Update smart_breed_matcher.py

smart_breed_matcher.py

@@ -18,18 +18,20 @@ def gpu_init_wrapper(func):
         return func(*args, **kwargs)
     return wrapper
 
-class SmartBreedMatcher:
-    def _safe_prediction(self, func):
-        @wraps(func)
-        def wrapper(*args, **kwargs):
-            try:
+def safe_prediction(func):
+    """錯誤處理裝飾器，提供 GPU 到 CPU 的降級機制"""
+    @wraps(func)
+    def wrapper(*args, **kwargs):
+        try:
+            return func(*args, **kwargs)
+        except RuntimeError as e:
+            if "CUDA" in str(e):
+                print("GPU 操作失敗，嘗試使用 CPU")
                 return func(*args, **kwargs)
-            except RuntimeError as e:
-                if "CUDA" in str(e):
-                    print("GPU 操作失敗，嘗試使用 CPU")
-                    return func(*args, **kwargs)
-                raise
-        return wrapper
+            raise
+    return wrapper
+
+class SmartBreedMatcher:
 
     def __init__(self, dog_data: List[Tuple]):
         self.dog_data = dog_data
@@ -891,7 +893,7 @@ class SmartBreedMatcher:
             }
 
     @gpu_init_wrapper
-    @_safe_prediction
+    @safe_prediction
     def match_user_preference(self, description: str, top_n: int = 10) -> List[Dict]:
         try:
             if self.model is None: