Update app.py (#2)

- Update app.py (5ab40f9df4716eeac234c343e89ab40c07f359f0)


Co-authored-by: Seungah Son <happysparkling@users.noreply.huggingface.co>

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):