Spaces:
Sleeping
Sleeping
Pratik Dwivedi
commited on
Commit
·
ca2a0f8
1
Parent(s):
f9b7868
sidebar fixes
Browse files
app.py
CHANGED
@@ -46,10 +46,10 @@ def main():
|
|
46 |
st.title("BetterZila RAG Enabled LLM")
|
47 |
with st.spinner("Registering Models for use..."):
|
48 |
prompter = register_gguf_model()
|
49 |
-
|
50 |
-
|
51 |
-
model_name = st.sidebar.selectbox("Select Model", prompter.model_catalog.models)
|
52 |
with st.spinner("Loading model..."):
|
|
|
53 |
prompter.load_model(model_name)
|
54 |
|
55 |
with st.spinner("Loading PDF content from the assignment URL..."):
|
|
|
46 |
st.title("BetterZila RAG Enabled LLM")
|
47 |
with st.spinner("Registering Models for use..."):
|
48 |
prompter = register_gguf_model()
|
49 |
+
|
50 |
+
model_name = st.sidebar.selectbox("Select Model", ["llama", "open_gpt4", "phi2", "mistral"])
|
|
|
51 |
with st.spinner("Loading model..."):
|
52 |
+
print(model_name)
|
53 |
prompter.load_model(model_name)
|
54 |
|
55 |
with st.spinner("Loading PDF content from the assignment URL..."):
|