Spaces:
Sleeping
Sleeping
Update pages/1_Simple_Chat_UI.py
Browse files
pages/1_Simple_Chat_UI.py
CHANGED
@@ -1,8 +1,6 @@
|
|
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"
|
@@ -21,7 +19,7 @@ def generate_response(prompt_input, model, tokenizer):
|
|
21 |
response = tokenizer.decode(outputs[0], skip_special_tokens=True).strip()
|
22 |
return response
|
23 |
|
24 |
-
st.set_page_config(page_title="Learn Geoscience")
|
25 |
|
26 |
with st.sidebar:
|
27 |
st.title('Learn Geoscience Chat')
|
|
|
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"
|
|
|
19 |
response = tokenizer.decode(outputs[0], skip_special_tokens=True).strip()
|
20 |
return response
|
21 |
|
22 |
+
st.set_page_config(page_title="Learn Geoscience", page_icon="../icon.png")
|
23 |
|
24 |
with st.sidebar:
|
25 |
st.title('Learn Geoscience Chat')
|