Soumen commited on
Commit
2e33caf
·
1 Parent(s): 0c8db44

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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 = 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)
 
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)