inputfield for model id

app.py

@@ -14,6 +14,7 @@ import torch
 
 gpt_model = 'gpt-4-1106-preview'
 embedding_model = 'text-embedding-3-small'
+default_model_id = "bigcode/starcoder2-7b"
 
 def init():
     if "conversation" not in st.session_state:
@@ -21,10 +22,8 @@ def init():
     if "chat_history" not in st.session_state:
         st.session_state.chat_history = None        
 
-def init_llm_pipeline():
-    if "llm" not in st.session_state:
-        model_id = "bigcode/starcoder2-7b"
-        
+def init_llm_pipeline(model_id):
+    if "llm" not in st.session_state:        
         tokenizer = AutoTokenizer.from_pretrained(model_id)
         model = AutoModelForCausalLM.from_pretrained(
             model_id,
@@ -97,11 +96,13 @@ def main():
 
 
     with st.sidebar:
+        st.subheader("Model selector")
+        model_id = st.text_input("Modelname on HuggingFace", default_model_id) 
         st.subheader("Code Upload")
         upload_docs=st.file_uploader("Dokumente hier hochladen", accept_multiple_files=True)
         if st.button("Hochladen"):
             with st.spinner("Analysiere Dokumente ..."):
-                init_llm_pipeline()
+                init_llm_pipeline(model_id)
                 raw_text = get_text(upload_docs)
                 vectorstore = get_vectorstore(raw_text)
                 st.session_state.conversation = get_conversation(vectorstore)