Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ import torch
|
|
3 |
from transformers import BertTokenizer, BertForMaskedLM
|
4 |
|
5 |
# Load the fine-tuned BERT model
|
6 |
-
model_name = "
|
7 |
tokenizer = BertTokenizer.from_pretrained(model_name)
|
8 |
model = BertForMaskedLM.from_pretrained(model_name)
|
9 |
model.to("cuda" if torch.cuda.is_available() else "cpu")
|
|
|
3 |
from transformers import BertTokenizer, BertForMaskedLM
|
4 |
|
5 |
# Load the fine-tuned BERT model
|
6 |
+
model_name = "fine_tuned_bert_model"
|
7 |
tokenizer = BertTokenizer.from_pretrained(model_name)
|
8 |
model = BertForMaskedLM.from_pretrained(model_name)
|
9 |
model.to("cuda" if torch.cuda.is_available() else "cpu")
|