Update pages/2_🆚_Extractive_vs_Abstractive.py

pages/2_🆚_Extractive_vs_Abstractive.py

@@ -25,10 +25,10 @@ def load_model():
     model_name = 'google/pegasus-billsum'
     torch_device = 'cuda' if torch.cuda.is_available() else 'cpu'
     #run using local model
-    tokenizer = PegasusTokenizer.from_pretrained(model_name)
-    model = PegasusForConditionalGeneration.from_pretrained(model_name, max_position_embeddings=2000).to(torch_device)
-    #tokenizer = PegasusTokenizer.from_pretrained("local_pegasus-billsum_tokenizer", use_auth_token=True)
-    #model = PegasusForConditionalGeneration.from_pretrained("local_pegasus-billsum_tokenizer_model", max_position_embeddings=2000, use_auth_token=True).to(torch_device)
+    #tokenizer = PegasusTokenizer.from_pretrained(model_name)
+    #model = PegasusForConditionalGeneration.from_pretrained(model_name, max_position_embeddings=2000).to(torch_device)
+    tokenizer = PegasusTokenizer.from_pretrained("local_pegasus-billsum_tokenizer")
+    model = PegasusForConditionalGeneration.from_pretrained("local_pegasus-billsum_tokenizer_model", max_position_embeddings=2000).to(torch_device)
     #tokenizer = PegasusTokenizer.from_pretrained("local_pegasus-billsum_tokenizer")
     #model = PegasusForConditionalGeneration.from_pretrained("local_pegasus-billsum_tokenizer_model", max_position_embeddings=2000).to(torch_device)
     return model,tokenizer