Spaces:
Paused
Paused
CountingMstar
commited on
Commit
•
b376f35
1
Parent(s):
07219dc
Update app.py
Browse files
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 |
|