movie_chatbot_v1

Model Card for Model ID

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

Model Details

• Model type: seq2seq(T5)
• License: MIT
• Finetuned from model: microsoft/GODEL-v1_1-base-seq2seq
• Final eval loss: 0.2331

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 [email protected]