Update app.py
Browse files
app.py
CHANGED
@@ -13,18 +13,18 @@ api_token = os.getenv("HUGGINGFACEHUB_API_TOKEN")
|
|
13 |
|
14 |
# Initialize model and tokenizer using the AutoModel from AirLLM
|
15 |
MAX_LENGTH = 128
|
16 |
-
model = AutoModel.from_pretrained("
|
17 |
|
18 |
# Streamlit app configuration
|
19 |
st.set_page_config(
|
20 |
-
page_title="Conversational Chatbot with
|
21 |
page_icon="🤖",
|
22 |
layout="wide",
|
23 |
initial_sidebar_state="expanded",
|
24 |
)
|
25 |
|
26 |
# App title
|
27 |
-
st.title("Conversational Chatbot with
|
28 |
|
29 |
# Sidebar configuration
|
30 |
st.sidebar.header("Chatbot Configuration")
|
@@ -91,6 +91,6 @@ if st.button("Send"):
|
|
91 |
st.sidebar.markdown(
|
92 |
"""
|
93 |
### About
|
94 |
-
This is a conversational chatbot built using the
|
95 |
"""
|
96 |
)
|
|
|
13 |
|
14 |
# Initialize model and tokenizer using the AutoModel from AirLLM
|
15 |
MAX_LENGTH = 128
|
16 |
+
model = AutoModel.from_pretrained("internlm/internlm2_5-7b-chat-gguf")
|
17 |
|
18 |
# Streamlit app configuration
|
19 |
st.set_page_config(
|
20 |
+
page_title="Conversational Chatbot with internlm2_5-7b-chat and AirLLM",
|
21 |
page_icon="🤖",
|
22 |
layout="wide",
|
23 |
initial_sidebar_state="expanded",
|
24 |
)
|
25 |
|
26 |
# App title
|
27 |
+
st.title("Conversational Chatbot with internlm2_5-7b-chat and AirLLM")
|
28 |
|
29 |
# Sidebar configuration
|
30 |
st.sidebar.header("Chatbot Configuration")
|
|
|
91 |
st.sidebar.markdown(
|
92 |
"""
|
93 |
### About
|
94 |
+
This is a conversational chatbot built using the internlm2_5-7b-chat model and AirLLM.
|
95 |
"""
|
96 |
)
|