Commit
·
de7c5ee
1
Parent(s):
a340b6b
Update app.py
Browse files
app.py
CHANGED
@@ -6,6 +6,7 @@ import string
|
|
6 |
bert_tokenizer = None
|
7 |
bert_model = None
|
8 |
top_k = 20
|
|
|
9 |
|
10 |
from transformers import BertTokenizer, BertForMaskedLM
|
11 |
|
@@ -125,6 +126,7 @@ def main():
|
|
125 |
global top_k
|
126 |
global bert_tokenizer
|
127 |
global bert_model
|
|
|
128 |
|
129 |
st.markdown("<h3 style='text-align: center;'>Qualitative evaluation of any pretrained BERT model</h3>", unsafe_allow_html=True)
|
130 |
st.markdown("""
|
|
|
6 |
bert_tokenizer = None
|
7 |
bert_model = None
|
8 |
top_k = 20
|
9 |
+
model_name = None
|
10 |
|
11 |
from transformers import BertTokenizer, BertForMaskedLM
|
12 |
|
|
|
126 |
global top_k
|
127 |
global bert_tokenizer
|
128 |
global bert_model
|
129 |
+
global model_name
|
130 |
|
131 |
st.markdown("<h3 style='text-align: center;'>Qualitative evaluation of any pretrained BERT model</h3>", unsafe_allow_html=True)
|
132 |
st.markdown("""
|