CountingMstar commited on
Commit
b376f35
1 Parent(s): 07219dc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from nltk.tokenize import word_tokenize
6
  import timm
7
 
8
  tokenizer = BertTokenizerFast.from_pretrained('bert-base-uncased')
9
- model = timm.create_model('hf_hub:pseudolab/AI_Tutor_BERT', pretrained=True)
10
  #model = BertForQuestionAnswering.from_pretrained("bert-base-uncased")
11
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
12
 
 
6
  import timm
7
 
8
  tokenizer = BertTokenizerFast.from_pretrained('bert-base-uncased')
9
+ # model = timm.create_model('hf_hub:pseudolab/AI_Tutor_BERT', pretrained=True)
10
  #model = BertForQuestionAnswering.from_pretrained("bert-base-uncased")
11
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
12