Update app.py
Browse files
app.py
CHANGED
@@ -9,14 +9,15 @@ st.markdown("### Articles classificator.")
|
|
9 |
@st.cache
|
10 |
def get_bert_and_tokenizer():
|
11 |
model_name = 'bert-base-uncased'
|
12 |
-
return AutoModel.from_pretrained(model_name), AutoTokenizer.from_pretrained(model_name)
|
|
|
13 |
|
14 |
-
|
15 |
|
16 |
class devops_model(nn.Module):
|
17 |
def __init__(self):
|
18 |
super(devops_model, self).__init__()
|
19 |
-
self.bert =
|
20 |
self.fc = nn.Sequential(
|
21 |
nn.Linear(768, 768),
|
22 |
nn.ReLU(),
|
|
|
9 |
@st.cache
|
10 |
def get_bert_and_tokenizer():
|
11 |
model_name = 'bert-base-uncased'
|
12 |
+
# return AutoModel.from_pretrained(model_name), AutoTokenizer.from_pretrained(model_name)
|
13 |
+
return AutoTokenizer.from_pretrained(model_name)
|
14 |
|
15 |
+
tokenizer = get_bert_and_tokenizer()
|
16 |
|
17 |
class devops_model(nn.Module):
|
18 |
def __init__(self):
|
19 |
super(devops_model, self).__init__()
|
20 |
+
self.bert = None
|
21 |
self.fc = nn.Sequential(
|
22 |
nn.Linear(768, 768),
|
23 |
nn.ReLU(),
|