Upload app.py

app.py

@@ -1,14 +1,13 @@
 import streamlit as st
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
-
+tokenizer = AutoTokenizer.from_pretrained('t5-base')
 @st.cache
 def load_model():
     model = AutoModelForSeq2SeqLM.from_pretrained(f"Supiri/t5-base-conversation")
-    tokenizer = AutoTokenizer.from_pretrained('t5-base')
     return model, tokenizer
 
-model, tokenizer = load_model()
+model = load_model()
 
 num_beams = st.slider('Number of beams', min_value=1, max_value=10, value=6)
 num_beam_groups = st.slider('Number of beam groups', min_value=1, max_value=10, value=2)