Deploy SafeChat ML service to Hugging Face Space

app/models/toxicity_classifier.py

@@ -423,7 +423,8 @@ class ToxicityClassifier:
         """Load MuRIL model into memory."""
         logger.info(f"Loading tokenizer and model ({self._model_name}) on {self.device}...")
         try:
-            self.tokenizer = AutoTokenizer.from_pretrained(self._model_name)
+            # MuRIL can require protobuf-backed slow tokenizer loading in lean container environments.
+            self.tokenizer = AutoTokenizer.from_pretrained(self._model_name, use_fast=False)
             self.model = AutoModelForSequenceClassification.from_pretrained(
                 self._model_name, 
                 num_labels=len(LABELS),

requirements.txt

@@ -8,6 +8,7 @@ transformers>=4.36.0
 tokenizers>=0.15.0
 accelerate>=0.25.0
 sentencepiece>=0.1.99
+protobuf>=4.25.3
 
 # --- FastAPI ---
 fastapi>=0.109.0