get model fix

app.py

@@ -7,14 +7,14 @@ import re
 from PIL import Image
 
 
-def get_model_gpt(model_name):
-    tokenizer = transformers.GPT2Tokenizer.from_pretrained(model_name)
+def get_model_gpt(model_name,tokenizer_name):
+    tokenizer = transformers.GPT2Tokenizer.from_pretrained(tokenizer_name)
     model = transformers.GPT2LMHeadModel.from_pretrained(model_name)
     model.eval()
     return model, tokenizer
 
-def get_model_t5(model_name):
-    tokenizer = transformers.T5Tokenizer.from_pretrained(model_name)
+def get_model_t5(model_name,tokenizer_name):
+    tokenizer = transformers.T5Tokenizer.from_pretrained(tokenizer_name)
     model = transformers.T5ForConditionalGeneration.from_pretrained(model_name)
     model.eval()
     return model, tokenizer