merve HF staff commited on
Commit
6463786
1 Parent(s): 0c8888d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -12
README.md CHANGED
@@ -21,15 +21,4 @@ Full credits to: [fchollet](https://twitter.com/fchollet)
21
  This example demonstrates how to implement text generation with a miniature GPT model. The model consists of a single Transformer block with causal masking in its attention layer.
22
 
23
  ## Datasets
24
- IMDB sentiment classification dataset for training. The model generates new movie reviews for a given prompt
25
-
26
- ## How to use
27
- You can use this model directly with a pipeline for text generation. Since the generation relies on some randomness, I
28
- set seed for reproducibility:
29
- ```python
30
- >>> from transformers import pipeline, set_seed
31
- >>> model = generation= pipeline('text-generation', model='keras-io/text-generation-miniature-gpt', tokenizer='bert-base-uncased')
32
- >>> set_seed(20)
33
- >>> generation("Once upon a time,", max_length=30, num_return_sequences=5)
34
-
35
- ```
 
21
  This example demonstrates how to implement text generation with a miniature GPT model. The model consists of a single Transformer block with causal masking in its attention layer.
22
 
23
  ## Datasets
24
+ IMDB sentiment classification dataset for training. The model generates new movie reviews for a given prompt.