Spaces:
Paused
Paused
Commit
•
3c61bbd
1
Parent(s):
e343140
Update app.py
Browse files
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("
|
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("pseudolab/AI_Tutor_BERT")
|
11 |
|
12 |
def get_prediction(context, question):
|
13 |
inputs = tokenizer.encode_plus(question, context, return_tensors='pt').to(device)
|