Update app.py

app.py

@@ -112,43 +112,46 @@ class ModelManager:
         return self._breed_model
 
     def _initialize_enhanced_system(self):
-        """Initialize enhanced multi-dimensional recommendation system"""
+        """Initialize enhanced multi-dimensional recommendation system (CPU-only load)"""
         if ModelManager._enhanced_system_initialized:
             return
-
+    
         try:
-            # Initialize SBERT model for semantic analysis
+            # Always load SBERT on CPU in the main process to avoid CUDA init
             try:
-                # Use default model configuration
-                model_name = 'all-MiniLM-L6-v2'
-                fallback_models = ['all-mpnet-base-v2', 'all-MiniLM-L12-v2']
-
-                for model_name_attempt in [model_name] + fallback_models:
+                model_options = [
+                    'sentence-transformers/all-MiniLM-L6-v2',
+                    'sentence-transformers/all-mpnet-base-v2',
+                    'sentence-transformers/all-MiniLM-L12-v2'
+                ]
+                self._sbert_model = None
+                for name in model_options:
                     try:
-                        self._sbert_model = SentenceTransformer(model_name_attempt)
-                        print(f"Initialized SBERT model: {model_name_attempt}")
+                        # CPU 載入，禁止在主行程碰 CUDA
+                        self._sbert_model = SentenceTransformer(name, device='cpu')
+                        print(f"Initialized SBERT model on CPU: {name}")
                         break
                     except Exception as e:
-                        print(f"Failed to load SBERT model {model_name_attempt}: {str(e)}")
+                        print(f"Failed to load SBERT model {name}: {e}")
                         continue
-
+    
                 if self._sbert_model is None:
-                    print("All SBERT models failed to load, enhanced system will use keyword-only analysis")
-
+                    print("All SBERT models failed; fallback to keyword-only analysis")
+    
             except Exception as e:
-                print(f"SBERT initialization failed: {str(e)}")
+                print(f"SBERT initialization failed: {e}")
                 self._sbert_model = None
-
+    
             ModelManager._enhanced_system_initialized = True
-            print("Enhanced recommendation system initialization completed")
-
+            print("Enhanced recommendation system initialization completed (CPU)")
+    
         except ImportError as e:
-            print(f"Enhanced modules not available: {str(e)}")
-            ModelManager._enhanced_system_initialized = True  # Mark as attempted
+            print(f"Enhanced modules not available: {e}")
+            ModelManager._enhanced_system_initialized = True
         except Exception as e:
-            print(f"Enhanced system initialization failed: {str(e)}")
+            print(f"Enhanced system initialization failed: {e}")
             print(traceback.format_exc())
-            ModelManager._enhanced_system_initialized = True  # Mark as attempted
+            ModelManager._enhanced_system_initialized = True
 
     @property
     def sbert_model(self):
@@ -630,4 +633,4 @@ def main():
 
 if __name__ == "__main__":
     iface = main()
-    iface.launch()
+    iface.launch()