This project generate The Office script using GPT-2 model.
You can download the model here: Hugging face
Fine tuning data: Kaggle
Base model: gpt-2 large
Epoch: 30
Train runtime: 6000.7391 secs
Loss: 0.0314
* First, Choose The Office character name.
* Second, Fill what the character will say in text. This will be base of script.
* And then, Fill number in length. Text is created as long as "length". I recommend between 100 and 300.
* If length is so big, generate time will be long.
name: The The Office character name.
text: The base of script.
length: The size of generated text. (min: 50)
{"0": [[character name, dialog], [character name, dialog], ...]}
curl -X POST "https://master-gpt2-the-office-fpem123.endpoint.ainize.ai/office" -H "accept: application/json" -H "Content-Type: multipart/form-data" -F "name=Michael" -F "text=Hello" -F "length=50"
{
"0": [
[
"Michael",
" Hello, good morning."
],
[
"Pam",
" Good morning."
],
[
"Michael",
" You ready? And we're off, like a herd of turtles."
],
[
"Michael",
" Well, Pam and I have eloped. Actually we just robbed a bank, and we are"
]
]
}
API page: Ainize
Demo page: End-point