Pratik Dwivedi commited on
Commit
ca2a0f8
·
1 Parent(s): f9b7868

sidebar fixes

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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
- # add a choice box for model selection to the sidebar
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..."):