Pratik Dwivedi commited on
Commit
9694dd7
·
1 Parent(s): ffb03cb
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -7,7 +7,6 @@ import PyPDF2
7
  def register_gguf_model():
8
 
9
  prompter = Prompt()
10
-
11
  your_model_name = "llama"
12
  hf_repo_name = "TheBloke/Llama-2-7B-Chat-GGUF"
13
  model_file = "llama-2-7b-chat.Q5_K_S.gguf"
@@ -36,7 +35,7 @@ def main():
36
 
37
  # keep the select box to llama as default but give a button right below it that says select model after which the model will be loaded
38
  st.sidebar.subheader("Select Model")
39
- model_name = st.sidebar.selectbox("Select Model", ["llama", "open_gpt4", "mistral", "phi2"])
40
  with st.spinner("Loading Model..."):
41
  prompter.load_model(model_name)
42
  st.success("Model Loaded!")
 
7
  def register_gguf_model():
8
 
9
  prompter = Prompt()
 
10
  your_model_name = "llama"
11
  hf_repo_name = "TheBloke/Llama-2-7B-Chat-GGUF"
12
  model_file = "llama-2-7b-chat.Q5_K_S.gguf"
 
35
 
36
  # keep the select box to llama as default but give a button right below it that says select model after which the model will be loaded
37
  st.sidebar.subheader("Select Model")
38
+ model_name = st.sidebar.selectbox("Select Model", ["llama", "open_gpt4", "phi2", "mistral"])
39
  with st.spinner("Loading Model..."):
40
  prompter.load_model(model_name)
41
  st.success("Model Loaded!")