Update model.py

model.py

@@ -43,7 +43,7 @@ np.random.seed(0)
 
 # find a better way to abstract the class
 class GPT2PPLV2:
-    def __init__(self, device="cpu", model_id="gpt2"):
+    def __init__(self, device="cpu", model_id="gpt2-medium"):
         self.device = device
         self.model_id = model_id
         self.model = GPT2LMHeadModel.from_pretrained(model_id).to(device)
@@ -54,7 +54,7 @@ class GPT2PPLV2:
         self.threshold = 0.7
 
         self.t5_model = transformers.AutoModelForSeq2SeqLM.from_pretrained("t5-base").to(device)
-        self.t5_tokenizer = T5Tokenizer.from_pretrained("t5-base", model_max_length=512)
+        self.t5_tokenizer = T5Tokenizer.from_pretrained("t5-larger", model_max_length=512)
 
     def apply_extracted_fills(self, masked_texts, extracted_fills):
         texts = []