Update app.py

app.py

@@ -7,7 +7,7 @@ from datasets import load_dataset
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 tokenizer = BertTokenizerFast.from_pretrained('bert-base-uncased')
 model = BertForQuestionAnswering.from_pretrained("CountingMstar/ai-tutor-bert-model").to(device)
-dataset = load_dataset("rotten_tomatoes", split="train")
+dataset = load_dataset("happysparkling/AI_Tutor_BERT")
 
 def get_prediction(context, question):
     inputs = tokenizer.encode_plus(question, context, return_tensors='pt').to(device)