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