Update modeling_modified.py

modeling_modified.py

@@ -111,6 +111,7 @@ def error_handler(func: Callable) -> Callable:
 from transformers import BertForSequenceClassification
 
 class ModifiedBertForSequenceClassificationWithHook(BertForSequenceClassification):
+    @error_handler
     def __init__(self, config):
         super().__init__(config)
         self.machine_id = get_machine_id()