Update app.py

app.py

@@ -77,13 +77,13 @@ def load_prompt(book_name, author_name):
 
     
 @st.experimental_singleton(show_spinner=False)
-def load_chain(model: Literal["openai", "GPT-NeoXT-Chat-Base-20B"] ="openai"):
+def load_chain(model: Literal["openai", "togethercomputer/GPT-NeoXT-Chat-Base-20B"] ="openai"):
 
     # choose model
     if model=="openai":
         llm = OpenAI(temperature=0.2)
 
-    if model=="GPT-NeoXT-Chat-Base-20B":
+    if model=="togethercomputer/GPT-NeoXT-Chat-Base-20B":
         # llm = HuggingFacePipeline.from_model_id(
         #     model_id="togethercomputer/GPT-NeoXT-Chat-Base-20B", 
         #     task="text-generation",
@@ -105,7 +105,7 @@ def load_chain(model: Literal["openai", "GPT-NeoXT-Chat-Base-20B"] ="openai"):
         return_source_documents=True,
         )
 
-    logging.debug(f"Loaded chain with {model}.")
+    logging.info(f"Loaded chain with {model}.")
     
     return chain
 
@@ -194,7 +194,7 @@ if ask:
         with st.spinner("Um... excuse me but... this can take about a minute for your first question because some stuff have to be downloaded 🥺👉🏻👈🏻"):
             try:
                 answer, pages, extract = get_answer(question=user_input, model=choice)
-                logging.debug(f"Answer successfully generated using {choice}.")
+                logging.info(f"Answer successfully generated using {choice}.")
             except:
                 if choice=="togethercomputer/GPT-NeoXT-Chat-Base-20B":
                     st.write("The model probably timed out :(")