Update web_demo-aiwei.py

2024-03-15 21:12:24 +08:00 · 2024-03-15 21:12:24 +08:00 · 5485b4c124
commit 5485b4c124
parent ccfa75c493
1 changed files with 2 additions and 0 deletions
--- a/web_demo-aiwei.py
+++ b/web_demo-aiwei.py
@ -210,6 +210,8 @@ def combine_history(prompt):


 def main():
+    st.markdown("我在这里，准备好倾听你的心声了。", unsafe_allow_html=True)
+    
    # torch.cuda.empty_cache()
    print("load model begin.")
    model, tokenizer = load_model()