Update src/core_model/predict.py

src/core_model/predict.py

@@ -9,22 +9,25 @@ class MindGuardPredictor:
         # 1. Identify the Model Hub ID
         self.model_id = "MohitRajput45/mindguard-xlmr"
         
-        # 2. Load the components
+        # 2. Identify the EXACT path inside that Repo
+        # Based on your HF screenshot, this is the full path to the files
+        hf_subfolder = "artifacts/xlmr_weights/final_mindguard_model"
+        
+        # 3. Load the components from the Hub
         try:
+            print(f"DEBUG: Attempting to load model from {self.model_id} in {hf_subfolder}...")
             self.tokenizer = XLMRobertaTokenizer.from_pretrained(
                 self.model_id, 
-                subfolder="final_mindguard_model"
+                subfolder=hf_subfolder
             )
             self.model = XLMRobertaForSequenceClassification.from_pretrained(
                 self.model_id, 
-                subfolder="final_mindguard_model"
+                subfolder=hf_subfolder
             )
-            print("✅ Model and Tokenizer loaded successfully from Hub.")
+            print("✅ Model and Tokenizer loaded successfully from Hub!")
         except Exception as e:
             print(f"❌ Error loading model: {e}")
-            # Fallback attempt if subfolder isn't present
-            self.tokenizer = XLMRobertaTokenizer.from_pretrained(self.model_id)
-            self.model = XLMRobertaForSequenceClassification.from_pretrained(self.model_id)
+            raise e # Stop the app if the model fails to load
         
         # 3. Emotion Mapping (Mathematical ID to English Word)
         self.emotion_map = {