Update models/hate_speech_classifier.py

models/hate_speech_classifier.py

@@ -7,11 +7,35 @@ import torch
 from deep_translator import GoogleTranslator
 
 class HateSpeechClassifier:
+    # def __init__(self):
+    #     base_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
+    #     models_dir = os.path.join(base_dir, "models", "model_weights", "custom_models")
+        
+    #     # Initialize translator
+    #     self.translator = GoogleTranslator(source='bn', target='en')
+        
+    #     # Use multiple pretrained models for better accuracy
+    #     self.pretrained_models = {
+    #         "primary": {
+    #             "name": "facebook/roberta-hate-speech-dynabench-r4-target",
+    #             "pipeline": None,
+    #             "weight": 0.6
+    #         },
+    #         "secondary": {
+    #             "name": "cardiffnlp/twitter-roberta-base-hate-latest",
+    #             "pipeline": None,
+    #             "weight": 0.4
+    #         }
+    #     }
     def __init__(self):
-        base_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
-        models_dir = os.path.join(base_dir, "models", "model_weights", "custom_models")
+        # Get absolute path to model weights
+        base_dir = Path(__file__).parent
+        self.model_dir = base_dir / 'model_weights' / 'custom_models'
         
-        # Initialize translator
+        # For Hugging Face Spaces, also check environment
+        if not self.model_dir.exists():
+            # Try alternative paths for deployed environment
+            self.model_dir = Path('/app/models/model_weights/custom_models')
         self.translator = GoogleTranslator(source='bn', target='en')
         
         # Use multiple pretrained models for better accuracy
@@ -28,6 +52,12 @@ class HateSpeechClassifier:
             }
         }
         
+        print(f"Model directory: {self.model_dir}")
+        print(f"Model directory exists: {self.model_dir.exists()}")
+        
+        if self.model_dir.exists():
+            print(f"Files in model directory: {list(self.model_dir.iterdir())}")
+        
         # English custom model paths
         self.english_model_path = os.path.join(models_dir, "english_model.pkl")
         self.english_vectorizer_path = os.path.join(models_dir, "english_vectorizer.pkl")