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 reddotbluename@gmail.com