not-lain commited on
Commit
ac746bc
1 Parent(s): fe60a88

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -9
app.py CHANGED
@@ -13,7 +13,7 @@ import time
13
  token = os.environ["HF_TOKEN"]
14
  ST = SentenceTransformer("mixedbread-ai/mxbai-embed-large-v1")
15
 
16
- dataset = load_dataset("not-lain/wikipedia",revision = "embedded")
17
 
18
  data = dataset["train"]
19
  data = data.add_faiss_index("embeddings") # column name that has the embeddings of the dataset
@@ -108,14 +108,7 @@ def talk(prompt,history):
108
  TITLE = "# RAG"
109
 
110
  DESCRIPTION = """
111
- A rag pipeline with a chatbot feature
112
-
113
- Resources used to build this project :
114
-
115
- * embedding model : https://huggingface.co/mixedbread-ai/mxbai-embed-large-v1
116
- * dataset : https://huggingface.co/datasets/not-lain/wikipedia
117
- * faiss docs : https://huggingface.co/docs/datasets/v2.18.0/en/package_reference/main_classes#datasets.Dataset.add_faiss_index
118
- * chatbot : https://huggingface.co/meta-llama/Meta-Llama-3-8B-Instruct
119
  """
120
 
121
 
 
13
  token = os.environ["HF_TOKEN"]
14
  ST = SentenceTransformer("mixedbread-ai/mxbai-embed-large-v1")
15
 
16
+ dataset = load_dataset("AI-4-Health/embedded-dataset")
17
 
18
  data = dataset["train"]
19
  data = data.add_faiss_index("embeddings") # column name that has the embeddings of the dataset
 
108
  TITLE = "# RAG"
109
 
110
  DESCRIPTION = """
111
+ HPP Chatbot
 
 
 
 
 
 
 
112
  """
113
 
114