Supiri commited on
Commit
125e91c
·
1 Parent(s): b61931d

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ tokenizer = AutoTokenizer.from_pretrained('t5-base')
5
  @st.cache
6
  def load_model():
7
  model = AutoModelForSeq2SeqLM.from_pretrained(f"Supiri/t5-base-conversation")
8
- return model, tokenizer
9
 
10
  model = load_model()
11
 
 
5
  @st.cache
6
  def load_model():
7
  model = AutoModelForSeq2SeqLM.from_pretrained(f"Supiri/t5-base-conversation")
8
+ return model
9
 
10
  model = load_model()
11