Update app.py

app.py

@@ -11,8 +11,6 @@ from langchain_core.runnables import RunnablePassthrough
 from langchain_core.output_parsers import StrOutputParser
 from transformers import AutoModel, AutoTokenizer, pipeline
 import joblib
-from huggingface_hub import hf_hub_download
-from llama_cpp import Llama
 from typing import List
 
 def setup_page():
@@ -195,7 +193,7 @@ def setup_rag_tab(llm):
     
     return chain
 
-def set_up_local_fine_tuned_tab(query):
+"""def setup_local_fine_tuned_tab(query):
     st.header("Fine-tuned Zephir model")
 
     llm = Llama(
@@ -212,9 +210,9 @@ def set_up_local_fine_tuned_tab(query):
             }
         ]
     )
-    return output["choices"][0]["message"]["content"]
+    return output["choices"][0]["message"]["content"]"""
 
-def setup_finetuned_tab():
+def setup_fine_tuned_tab():
     st.header("Fine-tuned Zephir model")
     
     llm = create_aws_ollama_llm()
@@ -314,7 +312,7 @@ with main_col:
                     st.code(soup.prettify(), language="xml-doc")"""
             """with tab4:
                 llm = create_aws_ollama_llm()
-                fine_tuned_chain = setup_finetuned_tab(llm)
+                fine_tuned_chain = setup_fine-tuned_tab(llm)
                 st.session_state.response_fine_tuned = fine_tuned_chain.invoke(query)
                 with st.chat_message("assistant"):
                     st.markdown(st.session_state.response_fine_tuned)"""