Happy hugging face

generator.py

@@ -35,12 +35,12 @@ import streamlit as st
 # hfmodel = pickle.load(open('models/hfmodel.sav', 'rb'))
 @st.cache
 def load_model():
-  hfmodel = alwm.from_pretrained("ThomasSimonini/t5-end2end-question-generation")
-  hftokenizer = T5TokenizerFast.from_pretrained("t5-base")
-  tokenizer = att.from_pretrained("ahotrod/albert_xxlargev1_squad2_512")
+  hfm = alwm.from_pretrained("ThomasSimonini/t5-end2end-question-generation")
+  hft = T5TokenizerFast.from_pretrained("t5-base")
+  tok = att.from_pretrained("ahotrod/albert_xxlargev1_squad2_512")
   model = amqa.from_pretrained("ahotrod/albert_xxlargev1_squad2_512")
-
-load_model()
+  return hfm, hft,tok, model
+hfmodel, hftokenizer, tokenizer, model = load_model()
 
 def run_model(input_string, **generator_args):
   generator_args = {