up

chatbot.py

@@ -309,23 +309,32 @@ Response:"""
         
     def _load_emotion_model(self):
         try:
+            # Use safetensors explicitly for the primary model
             return pipeline(
                 "text-classification",
                 model="SamLowe/roberta-base-go_emotions",
                 top_k=None,
                 device_map="auto" if self.device == "cuda" else None,
-                cache_dir=CACHE_DIR
+                cache_dir=CACHE_DIR,
+                use_safetensors=True  # Explicitly use safetensors
             )
         except Exception as e:
             logger.error(f"Error loading emotion model: {e}")
-            # Fallback 
-            return pipeline(
-                "text-classification", 
-                model="j-hartmann/emotion-english-distilroberta-base", 
-                return_all_scores=True,
-                device_map="auto" if self.device == "cuda" else None,
-                cache_dir=CACHE_DIR
-            )
+            # Fallback to a simpler model with safetensors
+            try:
+                return pipeline(
+                    "text-classification", 
+                    model="j-hartmann/emotion-english-distilroberta-base", 
+                    return_all_scores=True,
+                    device_map="auto" if self.device == "cuda" else None,
+                    cache_dir=CACHE_DIR,
+                    use_safetensors=True,  # Explicitly use safetensors
+                    from_tf=True  # Use TensorFlow weights if available
+                )
+            except Exception as e:
+                logger.error(f"Error loading fallback emotion model: {e}")
+                # Return a simple pipeline that always returns neutral
+                return lambda text: [{"label": "neutral", "score": 1.0}]
     
     def _initialize_llm(self, model_name: str, use_4bit: bool):
         try: