pvyas96 commited on
Commit
613da68
1 Parent(s): 20407e8

Update pages/1_Simple_Chat_UI.py

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