Generating pickle

generator.py

@@ -39,6 +39,9 @@ def load_model():
   hft = T5TokenizerFast.from_pretrained("t5-base")
   tok = att.from_pretrained("ahotrod/albert_xxlargev1_squad2_512")
   model = amqa.from_pretrained("ahotrod/albert_xxlargev1_squad2_512")
+  pickle.dump(model, open('model.sav', 'wb'))
+  pickle.dump(hfmodel, open('hfmodel.sav', 'wb'))
+
   return hfm, hft,tok, model
 hfmodel, hftokenizer, tokenizer, model = load_model()