Update README.md
Browse files
README.md
CHANGED
|
@@ -1,12 +1,7 @@
|
|
| 1 |
---
|
| 2 |
language: en
|
| 3 |
-
thumbnail: https://salesken.ai/assets/images/logo.png
|
| 4 |
license: apache-2.0
|
| 5 |
inference: false
|
| 6 |
-
widget:
|
| 7 |
-
- text: "every moment is a fresh beginning"
|
| 8 |
-
|
| 9 |
-
tags: salesken
|
| 10 |
---
|
| 11 |
|
| 12 |
Use this model to generate variations to augment the training data used for NLU systems.
|
|
@@ -19,8 +14,8 @@ if torch.cuda.is_available():
|
|
| 19 |
else :
|
| 20 |
device = "cpu"
|
| 21 |
|
| 22 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
| 23 |
-
model = AutoModelWithLMHead.from_pretrained("
|
| 24 |
|
| 25 |
input_query="every moment is a fresh beginning"
|
| 26 |
query= input_query + " ~~ "
|
|
|
|
| 1 |
---
|
| 2 |
language: en
|
|
|
|
| 3 |
license: apache-2.0
|
| 4 |
inference: false
|
|
|
|
|
|
|
|
|
|
|
|
|
| 5 |
---
|
| 6 |
|
| 7 |
Use this model to generate variations to augment the training data used for NLU systems.
|
|
|
|
| 14 |
else :
|
| 15 |
device = "cpu"
|
| 16 |
|
| 17 |
+
tokenizer = AutoTokenizer.from_pretrained("Ashishkr/Gpt2-paraphrase_generation")
|
| 18 |
+
model = AutoModelWithLMHead.from_pretrained("Ashishkr/Gpt2-paraphrase_generationn").to(device)
|
| 19 |
|
| 20 |
input_query="every moment is a fresh beginning"
|
| 21 |
query= input_query + " ~~ "
|