betterme
commited on
Commit
•
ca6142c
1
Parent(s):
d380ed4
update
Browse files- pages/chatbase.py +2 -2
pages/chatbase.py
CHANGED
|
@@ -10,7 +10,7 @@
|
|
| 10 |
|
| 11 |
from meutils.pipe import *
|
| 12 |
from chatllm.applications import ChatBase
|
| 13 |
-
from chatllm.utils import
|
| 14 |
|
| 15 |
import streamlit as st
|
| 16 |
from appzoo.streamlit_app.utils import display_pdf, reply4input
|
|
@@ -20,7 +20,7 @@ st.set_page_config('🔥ChatLLM', layout='centered', initial_sidebar_state='coll
|
|
| 20 |
|
| 21 |
@st.cache_resource
|
| 22 |
def get_chat_func():
|
| 23 |
-
chat_func =
|
| 24 |
return chat_func
|
| 25 |
|
| 26 |
|
|
|
|
| 10 |
|
| 11 |
from meutils.pipe import *
|
| 12 |
from chatllm.applications import ChatBase
|
| 13 |
+
from chatllm.utils import load_llm4chat
|
| 14 |
|
| 15 |
import streamlit as st
|
| 16 |
from appzoo.streamlit_app.utils import display_pdf, reply4input
|
|
|
|
| 20 |
|
| 21 |
@st.cache_resource
|
| 22 |
def get_chat_func():
|
| 23 |
+
chat_func = load_llm4chat("THUDM/chatglm-6b-int4")
|
| 24 |
return chat_func
|
| 25 |
|
| 26 |
|