Spaces:
Sleeping
Sleeping
Pratik Dwivedi
commited on
Commit
·
9694dd7
1
Parent(s):
ffb03cb
changes
Browse files
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", "
|
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!")
|