testin the trick

modeling_stacked.py

@@ -38,7 +38,7 @@ class SafeFloretWrapper(nn.Module):
 
     def forward(self, texts):
         # Floret expects strings, not tensors
-        _, predictions = self.model_floret.predict([texts], k=1)
+        _, predictions = self.floret_model.predict([texts], k=1)
         # Convert predictions to tensors for Hugging Face compatibility
         return torch.tensor(predictions)