Upload README.md
Browse files
README.md
CHANGED
@@ -44,9 +44,9 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
|
44 |
|
45 |
device = "cuda"
|
46 |
|
47 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
48 |
|
49 |
-
model = AutoModelForSeq2SeqLM.from_pretrained("
|
50 |
|
51 |
def paraphrase(
|
52 |
question,
|
|
|
44 |
|
45 |
device = "cuda"
|
46 |
|
47 |
+
tokenizer = AutoTokenizer.from_pretrained("kamal24/gpt_to_human")
|
48 |
|
49 |
+
model = AutoModelForSeq2SeqLM.from_pretrained("kamal24/gpt_to_human").to(device)
|
50 |
|
51 |
def paraphrase(
|
52 |
question,
|