Update app.py

app.py

@@ -4,6 +4,7 @@ from transformers import GPT2LMHeadModel, GPT2Tokenizer
 import torch
 
 ds = load_dataset("higgsfield/school-math-questions")
+qa_pairs = [(item['prompt'], item['completion']) for item in ds['train']]
 
 class MathDataset(torch.utils.data.Dataset):
     def __init__(self, qa_pairs, tokenizer, max_length=128):