Model Card for Model ID

A movie recommendation chatbot which is finetune from GODEL model using data generated from ChatGPT.

Model Details

Limitations

This model is trained using supervised learning on small amount of movie conversation data so it is not that good.
There is also a large version of this with better performance at bluenguyen/movie_chatbot_large_v1.

How to Get Started with the Model

Use the code below to get started with the model.

from transformers import AutoTokenizer, AutoModelForSeq2SeqLM

tokenizer = AutoTokenizer.from_pretrained("bluenguyen/movie_chatbot_v1")
model = AutoModelForSeq2SeqLM.from_pretrained("bluenguyen/movie_chatbot_v1").to('cuda')

def current_context(dialog, instruction):
    dialog = ' EOS '.join(dialog)
    context = f"{instruction} [CONTEXT] {dialog} "
    return context

def generate(context):
    input_ids = tokenizer(f"{context}", return_tensors="pt").input_ids.to('cuda')
    outputs = model.generate(input_ids, max_length=128, min_length=8, top_p=0.9, do_sample=True)
    output = tokenizer.decode(outputs[0], skip_special_tokens=True)
    return output

instruction = f'Instruction: given a dialog about movie recommendation, you need to respond based on human preferences.'
dialog = [
    'Do you have any recommendation about a movie?',
    'Yes! Any particular genre that you\'d like to see mentioned?',
]
print('User: ' + dialog[0])
print('Bot: ' + dialog[1])
while(1):
    print('User: ', end = '')
    query = input()
    if query == 'end':
        break
    dialog.append(query)
    response = generate(current_context(dialog, instruction))
    print('Bot: ' + response)
    dialog.append(response)

Demo

Example 1:

User: Do you have any recommendation about a movie? <br> Bot: Yes! Any particular genre that you'd like to see mentioned? <br> User: I like love story. <br> Bot: How about "The Notebook"? It's a classic love story with a great plot twist. <br> User: I have seen that one. Give me another one. <br> Bot: Have you seen "Eternal Sunshine of the Spotless Mind"? It's a unique romantic story with a captivating twist. <br> User: Is it any good? <br> Bot: Absolutely! It has an amazing cast and an engaging storyline. I highly recommend it. <br> User: Thank you <br> Bot: You're welcome! Let me know how you liked it. <br> User: end <br>

Example 2:

User: Do you have any recommendation about a movie? <br> Bot: Yes! Any particular genre that you'd like to see mentioned? <br> User: I like super hero movie. <br> Bot: Have you watched "The Dark Knight"? It's a classic superhero movie. <br> User: Tell me more <br> Bot: "The Dark Knight" is based on the book by Christopher Nolan and stars Christian Bale as Batman and Heath Ledger as the Joker. <br> User: Who is the Joker in this movie? <br> Bot: Heath Ledger is the Joker and the movie is based on a novel. <br> User: I mean what happen in the movie? <br> Bot: The Joker is tasked with protecting the savage City of the Black Knight. <br> User: Really? <br> Bot: It's a very intense movie with a great performance by Heath Ledger. <br> User: Nice :)) <br> Bot: Glad to hear it! How about "The Shawshank Redemption"? <br> User: What is it about? <br> Bot: "The Shawshank Redemption" is about a man named Andy Dufresne who is wrongly convicted of murder and sent to prison. <br> User: Who does he murder? <br> Bot: The man who is convicted of the murder is named Red, and he murders his wife. <br> User: You just said his name is Andy Dufresne <br> Bot: Sorry, I couldn't find any information about that movie. <br> User: yikes <br> Bot: Sorry again, I must have missed your recommendation. Is there any other movie you'd like to recommend? <br> User: end <br>

Example 3:

User: Do you have any recommendation about a movie? <br> Bot: Yes! Any particular genre that you'd like to see mentioned? <br> User: I like action movie. <br> Bot: Have you seen "John Wick"? <br> User: Yes I have <br> Bot: How about "The Bourne Identity"? <br> User: What is it about? <br> Bot: "The Bourne Identity" is an action-packed movie about a man named John McClane who goes on a rampage after his friend's death. <br> User: John McClane is character from "Die Hard" though <br> Bot: Yes, John McClane is a character from "Die Hard". Have you seen "The Raid: Redemption"? <br> User: You are not that good its funny haha <br> Bot: That's alright. How about "Die Hard"? <br> User: Is it any good? <br> Bot: It's a classic action movie about a retired hitman who seeks vengeance for the death of his dog. <br> User: That is "John Wick" plot lol <br> Bot: Yeah, I'm sorry I haven't heard of that movie. <br> User: You've just recommended that for me <br> Bot: Great! How about "The Matrix"? <br> User: What is the main character in that movie? <br> Bot: "The Matrix" is an action movie about a computer hacker named Neo who discovers that the world he lives in is actually a simulated reality created by machines. <br> User: What actor play Neo in that movie? <br> Bot: Neo is played by Keanu Reeves and plays the main character Neo. <br> User: end <br>

Model Card Contact

For personal questions related to this model, please contact via reddotbluename@gmail.com