pvyas96 commited on
Commit
aff3913
1 Parent(s): e57ea67

Update pages/1_Simple_Chat_UI.py

Browse files
Files changed (1) hide show
  1. pages/1_Simple_Chat_UI.py +1 -3
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')