Commit
·
7c8c08d
1
Parent(s):
167f69d
Update app.py
Browse files
app.py
CHANGED
@@ -4,8 +4,7 @@ import torch
|
|
4 |
import string
|
5 |
|
6 |
|
7 |
-
|
8 |
-
bert_model = None
|
9 |
|
10 |
from transformers import BertTokenizer, BertForMaskedLM
|
11 |
|
@@ -105,6 +104,8 @@ print(top_k)
|
|
105 |
|
106 |
# with st.spinner("Computing"):
|
107 |
try:
|
|
|
|
|
108 |
|
109 |
model_name = st.sidebar.selectbox(label='Select Model to Apply', options=['ajitrajasekharan/biomedical', 'bert-base-cased','bert-large-cased','microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract-fulltext','allenai/scibert_scivocab_cased'], index=0, key = "model_name")
|
110 |
option = st.selectbox(
|
|
|
4 |
import string
|
5 |
|
6 |
|
7 |
+
|
|
|
8 |
|
9 |
from transformers import BertTokenizer, BertForMaskedLM
|
10 |
|
|
|
104 |
|
105 |
# with st.spinner("Computing"):
|
106 |
try:
|
107 |
+
bert_tokenizer = None
|
108 |
+
bert_model = None
|
109 |
|
110 |
model_name = st.sidebar.selectbox(label='Select Model to Apply', options=['ajitrajasekharan/biomedical', 'bert-base-cased','bert-large-cased','microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract-fulltext','allenai/scibert_scivocab_cased'], index=0, key = "model_name")
|
111 |
option = st.selectbox(
|