Skip to content

Commit

Permalink
remove Feedback/Dashboard component from huggingface demo
Browse files Browse the repository at this point in the history
  • Loading branch information
jojortz committed Nov 22, 2023
1 parent 72e963f commit 071e791
Showing 1 changed file with 4 additions and 5 deletions.
9 changes: 4 additions & 5 deletions example/retrieval_qa/retrieval_qa_huggingface_demo.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,8 +20,8 @@
# "HuggingFaceH4/zephyr-7b-beta"
# "meta-llama/Llama-2-7b-chat-hf"
# "mistralai/Mistral-7B-v0.1"
# "databricks/dolly-v2-3b"
# "databricks/dolly-v2-3b"

RETRIEVAL_MODEL = os.getenv("RETRIEVAL_MODEL")


Expand Down Expand Up @@ -55,7 +55,7 @@ def main(**kwargs):
# retrieval, chatbot, and dashboard pykoi components
retriever = RetrievalQA(retrieval_model=retrieval_model, vector_db=vector_db, feedback="rag")
chatbot = Chatbot(None, feedback="rag", is_retrieval=True)
dashboard = Dashboard(RAGDatabase(), feedback="rag")
# dashboard = Dashboard(RAGDatabase(), feedback="rag")

############################################################
# Starting the application and retrieval qa as a component #
Expand All @@ -64,8 +64,7 @@ def main(**kwargs):
app = Application(debug=False, share=False)
app.add_component(retriever)
app.add_component(chatbot)
app.add_component(dashboard)
print("RUNNING APP IN DEMO MODE")
# app.add_component(dashboard)
app.run()


Expand Down

0 comments on commit 071e791

Please sign in to comment.