Update train_model.py

train_model.py

@@ -244,8 +244,8 @@ def main():
             logging_steps=500,
             learning_rate=5e-4,
             remove_unused_columns=False,
-            push_to_hub=False,  # We'll handle pushing manually
-            no_deprecation_warning=True  # Suppress FutureWarning
+            push_to_hub=False  # We'll handle pushing manually
+           
         )
     elif args.task == "classification":
         training_args = TrainingArguments(
@@ -258,8 +258,8 @@ def main():
             logging_steps=500,
             learning_rate=5e-5,
             remove_unused_columns=False,
-            push_to_hub=False,  # We'll handle pushing manually
-            no_deprecation_warning=True  # Suppress FutureWarning
+            push_to_hub=False  # We'll handle pushing manually
+            
         )
     else:
         logging.error("Unsupported task type for training arguments.")