Update README.md

README.md

@@ -13,50 +13,57 @@ widget:
 tags:
 - text-generation
 ---
-# SCRIPTGPT
-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.
+# ScriptGPT-small
 
-## Model description
-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.
+## 🖊️ Model description 
+ScriptGPT-small is a language model trained on a dataset of 100 YouTube videos that cover different domains of 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.
 
-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.
+The goal of ScriptGPT-small is to generate scripts for Youtube 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-small, 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.
 
-The current model is the smallest one with 124 million parameters (SRDdev/ScriptGPT-small)
+Models
+- [Script_GPT](https://huggingface.co/SRDdev/Script_GPT)      : AI content Model
+- [ScriptGPT-small](https://huggingface.co/SRDdev/ScriptGPT-small) : Generalized Content Model
 
 More models are coming soon...
 
-## Intended uses
-The intended uses of ScriptGPT include generating scripts for videos that explain artificial intelligence concepts, providing inspiration for content creators, and automating the process of generating video scripts.
+## 🛒 Intended uses
+The intended uses of ScriptGPT-small include generating scripts for videos, providing inspiration for content creators, and automating the process of generating video scripts. 
 
-## How to use
+
+## 📝 How to use
 You can use this model directly with a pipeline for text generation.
 
-__Load Model__
+1. __Load Model__
 ```python
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
 tokenizer = AutoTokenizer.from_pretrained("SRDdev/ScriptGPT-small")
 model = AutoModelForCausalLM.from_pretrained("SRDdev/ScriptGPT-small")
 ```
-__Pipeline__
+
+2. __Pipeline__
 ```python
 from transformers import pipeline
 generator = pipeline('text generation, model= model , tokenizer=tokenizer)
 
-context = "This is an introduction to Keras, a high-level neural networks API that is popular among researchers and developers. The video covers the basics of Keras, how to install it, and how to use it to build and train a neural network. The presenter demonstrates building and training a simple binary classification model using Keras"
-length_to_generate = 1000 
+context = "Cooking red sauce pasta"
+length_to_generate = 250 
 
 script = generator(context, max_length=length_to_generate, do_sample=True)[0]['generated_text']
 
 script
 ```
-Keeping the context more detailed results in better outputs
-
-## Limitations and bias
-The model is trained on Youtube Scripts and will work better for that. It may also generate random information and users should be aware of that and cross-validate the results.
+<p style="opacity: 0.8">The model may generate random information as it is still in beta version</p>
 
+## 🎈Limitations and bias
+> The model is trained on Youtube Scripts and will work better for that. It may also generate random information and users should be aware of that and cross-validate the results.
 
 ## Citations
+```
 @model{
         Name=Shreyas Dixit
         framework=Pytorch