Update train_llm.py

train_llm.py

@@ -19,12 +19,12 @@ data_directory = './fine_tune_data/'
 os.makedirs(data_directory, exist_ok=True)
 
 # Write the train data to a CSV file
-train_data='train_data.csv'
+train_data='train_data'
 train_filename = os.path.join(data_directory, train_data)
 dataset['train'].to_pandas().to_csv(train_filename, columns=['text'], index=False)
 
 # Write the validation data to a CSV file
-validation_data='validation_data.csv'
+validation_data='validation_data'
 validation_filename = os.path.join(data_directory, validation_data)
 dataset['validation'].to_pandas().to_csv(validation_filename, columns=['text'], index=False)