CountingMstar commited on
Commit
e343140
1 Parent(s): a657a9b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from datasets import load_dataset
7
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
8
  tokenizer = BertTokenizerFast.from_pretrained('bert-base-uncased')
9
  model = BertForQuestionAnswering.from_pretrained("CountingMstar/ai-tutor-bert-model").to(device)
10
- dataset = load_dataset("rotten_tomatoes", split="train")
11
 
12
  def get_prediction(context, question):
13
  inputs = tokenizer.encode_plus(question, context, return_tensors='pt').to(device)
 
7
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
8
  tokenizer = BertTokenizerFast.from_pretrained('bert-base-uncased')
9
  model = BertForQuestionAnswering.from_pretrained("CountingMstar/ai-tutor-bert-model").to(device)
10
+ dataset = load_dataset("happysparkling/AI_Tutor_BERT")
11
 
12
  def get_prediction(context, question):
13
  inputs = tokenizer.encode_plus(question, context, return_tensors='pt').to(device)