afnanmmir commited on
Commit
33f1f0d
1 Parent(s): 63245fe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -16,12 +16,12 @@ st_model_load = st.text('Loading summary generator model...')
16
  # @st.cache_data
17
  # def load_model():
18
  print("Loading model...")
19
- tokenizer = AutoTokenizer.from_pretrained(model_name)
20
- model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
21
- nltk.download('punkt')
22
- print("Model loaded!")
23
 
24
- tokenizer, model = load_model()
25
  st.success('Model loaded!')
26
  st_model_load.text("")
27
 
 
16
  # @st.cache_data
17
  # def load_model():
18
  print("Loading model...")
19
+ # tokenizer = AutoTokenizer.from_pretrained(model_name)
20
+ # model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
21
+ # nltk.download('punkt')
22
+ # print("Model loaded!")
23
 
24
+ # tokenizer, model = load_model()
25
  st.success('Model loaded!')
26
  st_model_load.text("")
27