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-medium"):
+    def __init__(self, device="cpu", model_id="gpt2"):
         self.device = device
         self.model_id = model_id
         self.model = GPT2LMHeadModel.from_pretrained(model_id).to(device)