Spaces:
Sleeping
Sleeping
Update pages/1_Simple_Chat_UI.py
Browse files
pages/1_Simple_Chat_UI.py
CHANGED
@@ -1,6 +1,8 @@
|
|
1 |
import streamlit as st
|
2 |
from transformers import AutoTokenizer, AutoModelForCausalLM, AutoModelForSeq2SeqLM
|
3 |
|
|
|
|
|
4 |
def load_model_tokenizer(model_name, hf_api_key):
|
5 |
if model_name == "LLaMa-2B":
|
6 |
model_name="llmware/bling-sheared-llama-2.7b-0.1"
|
|
|
1 |
import streamlit as st
|
2 |
from transformers import AutoTokenizer, AutoModelForCausalLM, AutoModelForSeq2SeqLM
|
3 |
|
4 |
+
st.set_page_config(page_title="Simple LLM Chatbot", page_icon="icon.png")
|
5 |
+
|
6 |
def load_model_tokenizer(model_name, hf_api_key):
|
7 |
if model_name == "LLaMa-2B":
|
8 |
model_name="llmware/bling-sheared-llama-2.7b-0.1"
|