Update app.py

app.py

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