From a0af67db690115117af2465a210a91e07d21fc75 Mon Sep 17 00:00:00 2001 From: Boyinapalli Sandeep Dora <61022690+LEAGUEDORA@users.noreply.github.com> Date: Sun, 9 Oct 2022 13:40:08 +0530 Subject: [PATCH] added use auth token according to new hugging face --- parrot/parrot.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/parrot/parrot.py b/parrot/parrot.py index f8ec738..41b56e2 100644 --- a/parrot/parrot.py +++ b/parrot/parrot.py @@ -1,14 +1,17 @@ +from typing import Text + + class Parrot(): - def __init__(self, model_tag="prithivida/parrot_paraphraser_on_T5", use_gpu=False): + def __init__(self, use_auth_token: Text, model_tag="prithivida/parrot_paraphraser_on_T5", use_gpu=False): from transformers import AutoTokenizer from transformers import AutoModelForSeq2SeqLM import pandas as pd from parrot.filters import Adequacy from parrot.filters import Fluency from parrot.filters import Diversity - self.tokenizer = AutoTokenizer.from_pretrained(model_tag) - self.model = AutoModelForSeq2SeqLM.from_pretrained(model_tag) + self.tokenizer = AutoTokenizer.from_pretrained(model_tag, use_auth_token = use_auth_token) + self.model = AutoModelForSeq2SeqLM.from_pretrained(model_tag, use_auth_token = use_auth_token) self.adequacy_score = Adequacy() self.fluency_score = Fluency() self.diversity_score= Diversity()