Sign inSign up

hitruong05/movie-chatbot

By hitruong05

•Updated over 1 year ago

Image
0

195

hitruong05/movie-chatbot repository overview

⁠Llama-2-chat-finetuned 🎬🤖

A fine-tuned Llama-2-7b-chat model optimized for movie-related conversations! This model understands movie genres, actors, directors, and can even recommend films based on your preferences.

⁠🚀 Features

  • Movie Q&A: Ask about movies, actors, directors, and genres.
  • Movie Recommendations: Get movie suggestions based on your interests.
  • Conversational AI: Engages in movie-related discussions.

⁠📌 Model Details

  • Base Model: NousResearch/Llama-2-7b-chat-hf⁠
  • Fine-Tuned By: HiTruong
  • Fine-Tuning Method: LoRA (Low-Rank Adaptation)
  • Dataset: Movie-related dataset
  • Evaluation Metric: BLEU Score
  • BLEU Score Before Fine-Tuning: 33.26
  • BLEU Score After Fine-Tuning: 77.53
  • Hosted On: Hugging Face 🤗⁠

⁠🐳 Deploy with Docker

⁠1️⃣ Pull the Docker Image
docker pull hitruong05/movie-chatbot
⁠2️⃣ Run the Container
docker run -p 7860:7860 hitruong05/movie-chatbot

The chatbot will now be accessible on

http://localhost:8501

⁠📌 About

  • Author: HiTruong
  • Built with: Python, Docker
  • License: MIT
  • Github Repo: Repo⁠

⁠📞 Contact

If you have any questions or suggestions, feel free to reach out:


⁠⭐ If you like this project, give it a star and share it! 🚀

Tag summary

Content type

Image

Digest

sha256:cec8a6b6c…

Size

3.4 GB

Last updated

over 1 year ago

docker pull hitruong05/movie-chatbot