Hugging Face's logo

SRDdev
/
ScriptForge-medium

Text Generation
Transformers
PyTorch
Safetensors
English
gpt2
text-generation-inference
Inference Endpoints
Model card Files Community
1
ScriptForge-medium / README.md
SRDdev's picture
SRDdev
Update README.md
90a26b0
preview code
|
raw
history blame contribute delete
2.91 kB
metadata
license: apache-2.0
language:
  - en
pipeline_tag: text-generation
widget:
  - text: 10 Meditation Tips
    example_title: Health Example
  - text: Cooking red sauce pasta
    example_title: Cooking Example
  - text: Introduction to Keras
    example_title: Technology Example
Tags:
  - text-generation
metrics:
  - accuracy

ScriptForge-medium

🖊️ Model description

ScriptForge-medium is a language model trained on a dataset of 250 YouTube videos that cover different domains of Youtube videos. ScriptForge-medium 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 ScriptForge-medium 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-medium, 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.

Models

More models are coming soon...

🛒 Intended uses

The intended uses of ScriptForge-medium include generating scripts for videos, providing inspiration for content creators, and automating the process of generating video scripts.

📝 How to use

You can use this model directly with a pipeline for text generation.

  1. Load Model
from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("SRDdev/ScriptForge-medium")
model = AutoModelForCausalLM.from_pretrained("SRDdev/ScriptForge-medium")
  1. Pipeline
from transformers import pipeline
generator = pipeline('text generation, model= model , tokenizer=tokenizer)

context = "Cooking red sauce pasta"
length_to_generate = 250 

script = generator(context, max_length=length_to_generate, do_sample=True)[0]['generated_text']

script

The model may generate random information as it is still in beta version

🎈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
        Year=Jan 2023
        Pipeline=text-generation
        Github=https://github.com/SRDdev
        LinkedIn=https://www.linkedin.com/in/srddev
      }