Edit model card
YAML Metadata Warning: empty or missing yaml metadata in repo card (https://huggingface.co/docs/hub/model-cards#model-card-metadata)

Tony Stark Chatbot

This model is a chatbot designed to generate responses in the style of Tony Stark, also known as Iron Man from the Marvel comics and movies.

Model Architecture

The tony_stark_chatbot is based on the DialoGPT-medium model, a powerful GPT-based architecture designed for generating conversational responses. It has been fine-tuned on a dataset of Tony Stark's dialogues from Marvel movie transcripts.

Usage

You can use this model to generate responses for a given input text using the following code:

from transformers import AutoModelForCausalLM, AutoTokenizer

tokenizer = AutoTokenizer.from_pretrained('diabolic6045/tony_stark_chatbot')
model = AutoModelForCausalLM.from_pretrained('diabolic6045/tony_stark_chatbot')

input_text = "What's your favorite hobby?"
input_tokens = tokenizer.encode(input_text, return_tensors='pt')
output_tokens = model.generate(input_tokens, max_length=50, num_return_sequences=1)
output_text = tokenizer.decode(output_tokens[0], skip_special_tokens=True)

print(output_text)

Limitations

This model is specifically designed to generate responses in the style of Tony Stark and may not provide accurate or coherent answers for general knowledge questions. It may also sometimes generate inappropriate responses. Be cautious while using this model in a public setting or for critical applications.

Training Data

The model was fine-tuned on a dataset of Tony Stark's dialogues from Marvel movie transcripts. The dataset was collected from publicly available movie scripts and includes conversations and quotes from various Marvel Cinematic Universe films.

Acknowledgments

This model was trained using the Hugging Face Transformers library, and it is based on the DialoGPT-medium model by Microsoft. Special thanks to the Hugging Face team and Microsoft for their contributions to the NLP community and Marvel for providing Dataset to train this model.


Feel free to test the model and provide feedback or report any issues. Enjoy chatting with Tony Stark!

Downloads last month
15
Safetensors
Model size
380M params
Tensor type
F32
·
BOOL
·