TomData commited on
Commit
d173e61
·
verified ·
1 Parent(s): 16d865e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +9 -9
README.md CHANGED
@@ -18,8 +18,7 @@ base_model: openai-community/gpt2-medium
18
  - **Model Type:** Transformer-based language model
19
  - **Language(s):** English
20
  - **Base Model:** [GPT2-medium](https://huggingface.co/openai-community/gpt2-medium)
21
- - **Resources for more information:**
22
- - [GitHub Repo](https://github.com/valentin-velev29/DLSS-24-GPT-2-Project)
23
 
24
 
25
  ## How to Get Started with the Model
@@ -29,10 +28,10 @@ set a seed for reproducibility:
29
 
30
  ```python
31
  >>> from transformers import pipeline, set_seed
32
- >>> generator = pipeline('text-generation', model='gpt2-medium')
33
  >>> set_seed(42)
34
  >>> generator("Hello, I'm a language model,", max_length=30, num_return_sequences=5)
35
-
36
 
37
 
38
  Here is how to use this model to get the features of a given text in PyTorch:
@@ -45,6 +44,7 @@ encoded_input = tokenizer(text, return_tensors='pt')
45
  output = model(**encoded_input)
46
  ```
47
 
 
48
  and in TensorFlow:
49
 
50
  ```python
@@ -58,11 +58,11 @@ output = model(encoded_input)
58
  ## Uses
59
 
60
  This model is further pretrained to generate artificial product reviews. This can be usefull for:
61
- > Market research
62
- > Product analysis
63
- > Customer preferences
64
- > Fashion trends
65
- > Research
66
 
67
 
68
  ## Training
 
18
  - **Model Type:** Transformer-based language model
19
  - **Language(s):** English
20
  - **Base Model:** [GPT2-medium](https://huggingface.co/openai-community/gpt2-medium)
21
+ - **Resources for more information:** [GitHub Repo](https://github.com/valentin-velev29/DLSS-24-GPT-2-Project)
 
22
 
23
 
24
  ## How to Get Started with the Model
 
28
 
29
  ```python
30
  >>> from transformers import pipeline, set_seed
31
+ >>> generator = pipeline('text-generation', model='TomData/GPT2-review')
32
  >>> set_seed(42)
33
  >>> generator("Hello, I'm a language model,", max_length=30, num_return_sequences=5)
34
+ ```
35
 
36
 
37
  Here is how to use this model to get the features of a given text in PyTorch:
 
44
  output = model(**encoded_input)
45
  ```
46
 
47
+
48
  and in TensorFlow:
49
 
50
  ```python
 
58
  ## Uses
59
 
60
  This model is further pretrained to generate artificial product reviews. This can be usefull for:
61
+ - Market research
62
+ - Product analysis
63
+ - Customer preferences
64
+ - Fashion trends
65
+ - Research
66
 
67
 
68
  ## Training