Team Ai
Modelpublic

SRDdev/ScriptForge-small

sourceHugging Faceapache-2.0updated 3y agoView on Hugging Face
1likes42downloads
README.md75 linesDownload Raw Back to root
1---2license: apache-2.03language:4- en5pipeline_tag: text-generation6widget:7- text: 10 Meditation tips8  example_title: Health Exmaple9- text: Cooking red sauce pasta10  example_title: Cooking Example11- text: Introduction to Keras12  example_title: Technology Example13tags:14- text-generation15---16# ScriptForge-small17 18## 🖊️ Model description 19ScriptForge-small is a language model trained on a dataset of 100 YouTube videos that cover different domains of Youtube videos.20ScriptForge-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. 21It generates a probability distribution over the next word given the previous words, without incorporating future words.22 23The goal of ScriptForge-small is to generate scripts for Youtube videos that are coherent, informative, and engaging. 24This can be useful for content creators who are looking for inspiration or who want to automate the process of generating video scripts. 25To 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.26 27Models28- [Script_GPT](https://huggingface.co/SRDdev/ScriptForge)      : AI content Model29- [ScriptGPT-small](https://huggingface.co/SRDdev/ScriptForge-small) : Generalized Content Model30 31More models are coming soon...32 33## 🛒 Intended uses34The intended uses of ScriptForge-small include generating scripts for videos, providing inspiration for content creators, and automating the process of generating video scripts. 35 36 37## 📝 How to use38You can use this model directly with a pipeline for text generation.39 401. __Load Model__41```python42from transformers import AutoTokenizer, AutoModelForCausalLM43 44tokenizer = AutoTokenizer.from_pretrained("SRDdev/ScriptForge-small")45model = AutoModelForCausalLM.from_pretrained("SRDdev/ScriptForge-small")46```47 482. __Pipeline__49```python50from transformers import pipeline51generator = pipeline('text generation, model= model , tokenizer=tokenizer)52 53context = "Cooking red sauce pasta"54length_to_generate = 250 55 56script = generator(context, max_length=length_to_generate, do_sample=True)[0]['generated_text']57 58script59```60<p style="opacity: 0.8">The model may generate random information as it is still in beta version</p>61 62## 🎈Limitations and bias63> 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.64 65## Citations66```67@model{68        Name=Shreyas Dixit69        framework=Pytorch70        Year=Jan 202371        Pipeline=text-generation72        Github=https://github.com/SRDdev73        LinkedIn=https://www.linkedin.com/in/srddev74      }75```