Spaces:
Runtime error
Runtime error
| import os | |
| os.system("pip install transformers datasets torch gradio") # Install missing libraries | |
| from transformers import Trainer, TrainingArguments | |
| # Rest of your code... | |
| from transformers import Trainer, TrainingArguments | |
| training_args = TrainingArguments( | |
| output_dir="./results", | |
| num_train_epochs=3, | |
| per_device_train_batch_size=8, | |
| save_steps=10_000, | |
| save_total_limit=2, | |
| ) | |
| trainer = Trainer( | |
| model=model, | |
| args=training_args, | |
| train_dataset="training_data.jsonl", | |
| ) | |
| trainer.train() | |