Update smart_breed_matcher.py

smart_breed_matcher.py

@@ -1,6 +1,7 @@
 import torch
 import re
 import numpy as np
+import spaces
 from typing import List, Dict, Tuple, Optional
 from dataclasses import dataclass
 from breed_health_info import breed_health_info
@@ -8,6 +9,15 @@ from breed_noise_info import breed_noise_info
 from dog_database import dog_data
 from scoring_calculation_system import UserPreferences
 from sentence_transformers import SentenceTransformer, util
+from functools import wraps
+
+def gpu_init_wrapper(func):
+    @spaces.GPU
+    @wraps(func)
+    def wrapper(*args, **kwargs):
+        return func(*args, **kwargs)
+    return wrapper
+
 
 class SmartBreedMatcher:
     def __init__(self, dog_data: List[Tuple]):
@@ -857,6 +867,7 @@ class SmartBreedMatcher:
                 'good_with_children': False
             }
 
+    @gpu_init_wrapper
     def match_user_preference(self, description: str, top_n: int = 10) -> List[Dict]:
         try:
             # 獲取場景權重