pt-sk commited on
Commit
a74b7ca
1 Parent(s): 0c74f96

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -16
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).