cahya's picture
update the description
eb0cecf
import streamlit as st
import codecs
import streamlit.components.v1 as stc
import pathlib
def stc_chatbot(root_dir, width=700, height=600):
html_file = root_dir/"app/chatbot.html"
css_file = root_dir/"app/css/chatbot.css"
js_file = root_dir/"app/js/chatbot.js"
if css_file.exists() and js_file.exists():
html = codecs.open(html_file, "r").read()
css = codecs.open(css_file, "r").read()
js = codecs.open(js_file, "r").read()
html = html.replace('<link rel="stylesheet" href="css/chatbot.css">', "<style>\n" + css + "\n</style>")
html = html.replace('<script src="js/chatbot.js"></script>', "<script>\n" + js + "\n</script>")
stc.html(html, width=width, height=height, scrolling=False)
st.header("English and Indonesian Persona Chatbot")
description = f"This application uses the Indonesian GPT2 model; we finetuned it with the original English persona " \
f"dataset and its Indonesian translation. It gives the chatbot the capability to understand and talk " \
f"in both languages (code-switching). The finetuning is based on the Huggingface's Conversational AI " \
f"with Transfer Learning."
st.markdown(description)
root_dir = pathlib.Path(".")
stc_chatbot(root_dir)