update available sources

towardsai · Jul 27, 2024 · dde771e · dde771e
1 parent 187d389
commit dde771e
Showing 1 changed file with 13 additions and 9 deletions.
diff --git a/scripts/gradio-ui.py b/scripts/gradio-ui.py
@@ -50,21 +50,25 @@
 
 AVAILABLE_SOURCES_UI = [
     "HF Transformers",
+    "PEFT",
+    "TRL",
+    "LlamaIndex Docs",
     "Towards AI Blog",
-    "Wikipedia",
-    "OpenAI Docs",
-    "LangChain Docs",
-    "LLama-Index Docs",
+    # "Wikipedia",
+    # "OpenAI Docs",
+    # "LangChain Docs",
     "RAG Course",
 ]
 
 AVAILABLE_SOURCES = [
     "HF_Transformers",
+    "PEFT",
+    "TRL",
+    "LlamaIndex",
     "towards_ai_blog",
-    "wikipedia",
-    "openai_docs",
-    "langchain_docs",
-    "llama_index_docs",
+    # "wikipedia",
+    # "openai_docs",
+    # "langchain_docs",
     "rag_course",
 ]
 
@@ -360,7 +364,7 @@ def vote(data: gr.LikeData):
 
 accordion = gr.Accordion(label="Customize Sources (Click to expand)", open=False)
 sources = gr.CheckboxGroup(
-    AVAILABLE_SOURCES_UI, label="Sources", value="HF Transformers", interactive=False  # type: ignore
+    AVAILABLE_SOURCES_UI, label="Sources", value=["HF Transformers", "PEFT", "TRL", "LlamaIndex Docs"], interactive=False  # type: ignore
 )
 model = gr.Dropdown(
     [