Update app.py
Browse files
app.py
CHANGED
|
@@ -98,8 +98,8 @@ def load_models():
|
|
| 98 |
tokenizer = AutoTokenizer.from_pretrained('gpt2-large')
|
| 99 |
model = GPT2LMHeadModel.from_pretrained('gpt2-large')
|
| 100 |
return tokenizer, model
|
| 101 |
-
tokenizer = AutoTokenizer.from_pretrained('t5-base')
|
| 102 |
-
model = AutoModelWithLMHead.from_pretrained('t5-base', return_dict=True)
|
| 103 |
# Function For Extracting Entities
|
| 104 |
@st.experimental_singleton
|
| 105 |
#@st.cache_resource(experimental_allow_widgets=True)
|
|
|
|
| 98 |
tokenizer = AutoTokenizer.from_pretrained('gpt2-large')
|
| 99 |
model = GPT2LMHeadModel.from_pretrained('gpt2-large')
|
| 100 |
return tokenizer, model
|
| 101 |
+
global tokenizer = AutoTokenizer.from_pretrained('t5-base')
|
| 102 |
+
global model = AutoModelWithLMHead.from_pretrained('t5-base', return_dict=True)
|
| 103 |
# Function For Extracting Entities
|
| 104 |
@st.experimental_singleton
|
| 105 |
#@st.cache_resource(experimental_allow_widgets=True)
|