Upload train.py

train.py

@@ -0,0 +1,32 @@
+from transformers import AutoModel, AutoTokenizer, Trainer, TrainingArguments
+from datasets import load_dataset
+import pandas as pd
+
+# Load model and tokenizer
+model = AutoModel.from_pretrained('Anime.safetensors', trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained('Saad381/SpectraGen')
+
+# Load dataset
+dataset = load_dataset('csv', data_files='dataset.csv')
+
+# Set training arguments
+training_args = TrainingArguments(
+    output_dir='./results',
+    per_device_train_batch_size=8,
+    num_train_epochs=3,
+    evaluation_strategy="epoch"
+)
+
+# Initialize Trainer
+trainer = Trainer(
+    model=model,
+    args=training_args,
+    train_dataset=dataset['train']
+)
+
+# Train the model
+trainer.train()
+
+# Save the model
+model.save_pretrained('./trained_model')
+tokenizer.save_pretrained('./trained_model')