Update app.py

app.py

@@ -3,9 +3,11 @@ from transformers import BertForQuestionAnswering
 from transformers import BertTokenizerFast
 import torch
 from nltk.tokenize import word_tokenize
+import timm
 
 tokenizer = BertTokenizerFast.from_pretrained('bert-base-uncased')
-model = BertForQuestionAnswering.from_pretrained("bert-base-uncased")
+model = timm.create_model('hf_hub:pseudolab/AI_Tutor_BERT', pretrained=True)
+#model = BertForQuestionAnswering.from_pretrained("bert-base-uncased")
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 def get_prediction(context, question):