Update README.md

README.md

@@ -54,6 +54,7 @@ Use the code below to get started with the model.
 
 ### Training Data
 
+```
 from transformers import GPT2Tokenizer
 
 dataset = load_dataset("FinGPT/fingpt-sentiment-train")
@@ -99,9 +100,11 @@ tokenized_datasets['train'] = train_test_split['train']
 
 small_train_dataset = tokenized_datasets["train"].shuffle(seed=42).select(range(100))
 small_eval_dataset = tokenized_datasets["test"].shuffle(seed=42).select(range(100))
+```
 
 ### Fine-tune Procedure 
 
+```
 from transformers import GPT2ForSequenceClassification
 from transformers import TrainingArguments, Trainer
 
@@ -126,6 +129,7 @@ trainer = Trainer(
 trainer.train()
 trainer.evaluate()
 trainer.save_model("fine_tuned_finsetiment_model")
+```
 
 #### Training Hyperparameters
 
@@ -139,6 +143,7 @@ trainer.save_model("fine_tuned_finsetiment_model")
 
 ## Evaluation
 
+```
 import evaluate
 
 metric = evaluate.load("accuracy")
@@ -148,6 +153,7 @@ def compute_metrics(eval_pred):
    predictions = np.argmax(logits, axis=-1)
 
    return metric.compute(predictions=predictions, references=labels)
+```
 
 ### Results