Update README.md
Browse files
README.md
CHANGED
@@ -1,19 +1,4 @@
|
|
1 |
---
|
2 |
license: mit
|
3 |
---
|
4 |
-
GPT2-IMDB is trained on IMDB dataset to generate a review for the given movie name(generates positive and negative reviews).
|
5 |
-
|
6 |
-
```py
|
7 |
-
# Use a pipeline as a high-level helper
|
8 |
-
from transformers import pipeline
|
9 |
-
|
10 |
-
pipe = pipeline("text-generation", model="pt-sk/gpt2-imdb")
|
11 |
-
```
|
12 |
-
|
13 |
-
```py
|
14 |
-
# Load model directly
|
15 |
-
from transformers import AutoTokenizer, AutoModelForCausalLM
|
16 |
-
|
17 |
-
tokenizer = AutoTokenizer.from_pretrained("pt-sk/gpt2-imdb")
|
18 |
-
model = AutoModelForCausalLM.from_pretrained("pt-sk/gpt2-imdb")
|
19 |
-
```
|
|
|
1 |
---
|
2 |
license: mit
|
3 |
---
|
4 |
+
GPT2-IMDB is trained on IMDB dataset to generate a review for the given movie name(generates positive and negative reviews).
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|