w3robotics commited on
Commit
e40b8c7
·
verified ·
1 Parent(s): 86010e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -6
app.py CHANGED
@@ -8,8 +8,16 @@ from langchain_core.prompts import PromptTemplate
8
  from langchain_core.output_parsers import StrOutputParser
9
 
10
 
 
 
 
 
 
 
 
11
  # Variables used Globally
12
  path = "/data" #preset path for hugging face spaces for persistent storage and cannot be changed
 
13
 
14
  # Application Functions
15
  # File Loader
@@ -128,7 +136,6 @@ with recycle_column:
128
  # Main app goes below here -
129
 
130
 
131
- model_id="mistralai/Mistral-7B-Instruct-v0.3"
132
 
133
  def get_llm_hf_inference(model_id=model_id, max_new_tokens=128, temperature=0.1):
134
  """
@@ -150,11 +157,6 @@ def get_llm_hf_inference(model_id=model_id, max_new_tokens=128, temperature=0.1)
150
  )
151
  return llm
152
 
153
- # Configure the Streamlit app
154
- st.set_page_config(page_title="HuggingFace ChatBot", page_icon="🤗")
155
- st.title("Personal HuggingFace ChatBot")
156
- st.markdown(f"*This is a simple chatbot that uses the HuggingFace transformers library to generate responses to your text input. It uses the {model_id}.*")
157
-
158
  # Initialize session state for avatars
159
  if "avatars" not in st.session_state:
160
  st.session_state.avatars = {'user': None, 'assistant': None}
 
8
  from langchain_core.output_parsers import StrOutputParser
9
 
10
 
11
+ # Configure the Streamlit app
12
+ st.set_page_config(page_title="HuggingFace ChatBot", page_icon="🤗")
13
+ st.title("Personal HuggingFace ChatBot")
14
+ st.markdown(f"*This is a simple chatbot that uses the HuggingFace transformers library to generate responses to your text input. It uses the {model_id}.*")
15
+
16
+
17
+
18
  # Variables used Globally
19
  path = "/data" #preset path for hugging face spaces for persistent storage and cannot be changed
20
+ model_id="mistralai/Mistral-7B-Instruct-v0.3"
21
 
22
  # Application Functions
23
  # File Loader
 
136
  # Main app goes below here -
137
 
138
 
 
139
 
140
  def get_llm_hf_inference(model_id=model_id, max_new_tokens=128, temperature=0.1):
141
  """
 
157
  )
158
  return llm
159
 
 
 
 
 
 
160
  # Initialize session state for avatars
161
  if "avatars" not in st.session_state:
162
  st.session_state.avatars = {'user': None, 'assistant': None}