zenith04's picture
Update app.py
2065852 verified
import streamlit as st
import pandas as pd
import pickle
#------------------------------------RAG BASED CHATBOT ---------------------------------------
import sys
import os
sys.path.append(os.path.abspath("src"))
from recommendationSystem.chatbot.client_module.utils import chatbot
# βœ… Redirect model cache and config to writable path
os.environ["HF_HOME"] = "/tmp/huggingface"
os.environ["TRANSFORMERS_CACHE"] = "/tmp/transformers"
os.environ["SENTENCE_TRANSFORMERS_HOME"] = "/tmp/sentence-transformers"
chatbot()
#render_history_download()
#----------------------------------- RECOMMEDATION SYSTEM -----------------------------------------
from utils import fetch_transformed_data, anime_info
st.title("Anime Recommender System")
data_path, matrix_path = fetch_transformed_data()
anime_data = pd.read_csv(data_path)
similarity_matrix = pickle.load(open(file=matrix_path,mode='rb'))
select_anime_name = st.selectbox(
"Choose Anime Name : ",
anime_data['title'].values,
index=None,
placeholder="Select the anime for recommendation..."
)
anime_info(anime_name=select_anime_name,anime_data=anime_data,similarity_matrix=similarity_matrix)