Update generator.py

generator.py

@@ -7,7 +7,7 @@ from transformers import (
     AutoModelForQuestionAnswering as amqa,
     AutoTokenizer as att,
     # BertTokenizer,
-    # AlbertTokenizer,
+    AlbertTokenizer,
     # BertForQuestionAnswering,
     # AlbertForQuestionAnswering,
     # T5Config,
@@ -37,7 +37,7 @@ def load_model():
   hfm = pickle.load(open('hfmodel.sav','rb'))
   hft = T5TokenizerFast.from_pretrained("t5-base")
   model = pickle.load(open('model.sav','rb'))
-  tok = att.from_pretrained("ahotrod/albert_xxlargev1_squad2_512 ")
+  tok = AlbertTokenizer.from_pretrained("ahotrod/albert_xxlargev1_squad2_512")
   # return hfm, hft,tok, model
   return hfm, hft,tok, model