Update README.md
Browse files
README.md
CHANGED
@@ -13,11 +13,11 @@ tags:
|
|
13 |
Pretrained model on the English language using a causal language modeling (CLM) objective. It was introduced in this paper and first released on this page.
|
14 |
|
15 |
## Model description
|
16 |
-
ScriptGPT is a language model trained on a dataset of CUSTOM YouTube videos.
|
17 |
|
18 |
The goal of ScriptGPT is to generate scripts for AI videos that are coherent, informative, and engaging. This can be useful for content creators who are looking for inspiration or who want to automate the process of generating video scripts. To use ScriptGPT, users can provide a prompt or a starting sentence, and the model will generate a sequence of words that follow the context and style of the training data.
|
19 |
|
20 |
-
The current model is the smallest one with 124 million parameters (SRDdev/
|
21 |
|
22 |
More models are coming soon...
|
23 |
|
@@ -31,8 +31,8 @@ __Load Model__
|
|
31 |
```python
|
32 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
33 |
|
34 |
-
tokenizer = AutoTokenizer.from_pretrained("SRDdev/
|
35 |
-
model = AutoModelForCausalLM.from_pretrained("SRDdev/
|
36 |
```
|
37 |
__Pipeline__
|
38 |
```python
|
|
|
13 |
Pretrained model on the English language using a causal language modeling (CLM) objective. It was introduced in this paper and first released on this page.
|
14 |
|
15 |
## Model description
|
16 |
+
ScriptGPT is a language model trained on a dataset of CUSTOM YouTube videos. ScriptGPT-small is a Causal language transformer. The model resembles the GPT2 architecture, the model is a Causal Language model meaning it predicts the probability of a sequence of words based on the preceding words in the sequence. It generates a probability distribution over the next word given the previous words, without incorporating future words.
|
17 |
|
18 |
The goal of ScriptGPT is to generate scripts for AI videos that are coherent, informative, and engaging. This can be useful for content creators who are looking for inspiration or who want to automate the process of generating video scripts. To use ScriptGPT, users can provide a prompt or a starting sentence, and the model will generate a sequence of words that follow the context and style of the training data.
|
19 |
|
20 |
+
The current model is the smallest one with 124 million parameters (SRDdev/ScriptGPT-small)
|
21 |
|
22 |
More models are coming soon...
|
23 |
|
|
|
31 |
```python
|
32 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
33 |
|
34 |
+
tokenizer = AutoTokenizer.from_pretrained("SRDdev/ScriptGPT-small")
|
35 |
+
model = AutoModelForCausalLM.from_pretrained("SRDdev/ScriptGPT-small")
|
36 |
```
|
37 |
__Pipeline__
|
38 |
```python
|