Spaces:
Runtime error
Runtime error
oryx1729
commited on
Commit
·
8df121c
1
Parent(s):
91eb887
Add indexing
Browse files- app.py +15 -21
- faiss_document_store.db +0 -0
- my_faiss_config.json +1 -0
- my_faiss_index.faiss +0 -0
- requirements.txt +1 -1
app.py
CHANGED
|
@@ -1,12 +1,9 @@
|
|
|
|
|
| 1 |
import logging
|
| 2 |
import sys
|
| 3 |
|
| 4 |
import streamlit as st
|
| 5 |
-
from haystack import Document
|
| 6 |
from haystack import Pipeline
|
| 7 |
-
from haystack.document_stores import InMemoryDocumentStore
|
| 8 |
-
from haystack.nodes import EmbeddingRetriever
|
| 9 |
-
from haystack.nodes import FARMReader
|
| 10 |
|
| 11 |
logging.basicConfig(
|
| 12 |
level=logging.DEBUG,
|
|
@@ -17,29 +14,26 @@ logging.basicConfig(
|
|
| 17 |
|
| 18 |
p = None
|
| 19 |
|
|
|
|
| 20 |
def app_init():
|
| 21 |
-
|
| 22 |
-
|
| 23 |
-
|
| 24 |
-
ds
|
| 25 |
-
|
| 26 |
-
|
| 27 |
-
|
| 28 |
-
|
| 29 |
-
|
| 30 |
-
)
|
| 31 |
-
ds.update_embeddings(retriever)
|
| 32 |
-
reader = FARMReader("deepset/minilm-uncased-squad2", use_gpu=False)
|
| 33 |
-
p = Pipeline()
|
| 34 |
-
p.add_node(component=retriever, name='retriever', inputs=['Query'])
|
| 35 |
-
p.add_node(component=reader, name='reader', inputs=['retriever'])
|
| 36 |
|
| 37 |
|
| 38 |
def main():
|
| 39 |
app_init()
|
| 40 |
st.title("Haystack Demo")
|
| 41 |
input = st.text_input("Query ...")
|
| 42 |
-
st.text(p.run(input))
|
|
|
|
| 43 |
|
| 44 |
if __name__ == "__main__":
|
| 45 |
-
main()
|
|
|
|
| 1 |
+
import glob
|
| 2 |
import logging
|
| 3 |
import sys
|
| 4 |
|
| 5 |
import streamlit as st
|
|
|
|
| 6 |
from haystack import Pipeline
|
|
|
|
|
|
|
|
|
|
| 7 |
|
| 8 |
logging.basicConfig(
|
| 9 |
level=logging.DEBUG,
|
|
|
|
| 14 |
|
| 15 |
p = None
|
| 16 |
|
| 17 |
+
|
| 18 |
def app_init():
|
| 19 |
+
indexing_pipeline = Pipeline.load_from_yaml("pipeline.yaml", pipeline_name="indexing")
|
| 20 |
+
file_paths = glob.glob("data/*")
|
| 21 |
+
ds = indexing_pipeline.get_node("DocumentStore")
|
| 22 |
+
ds.delete_all_documents()
|
| 23 |
+
indexing_pipeline.run(file_paths=file_paths)
|
| 24 |
+
ds.update_embeddings(indexing_pipeline.get_node("Retriever"))
|
| 25 |
+
ds.save(config_path="my_faiss_config.json", index_path="my_faiss_index.faiss")
|
| 26 |
+
|
| 27 |
+
global p
|
| 28 |
+
p = Pipeline.load_from_yaml("pipeline.yaml", pipeline_name="query")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 29 |
|
| 30 |
|
| 31 |
def main():
|
| 32 |
app_init()
|
| 33 |
st.title("Haystack Demo")
|
| 34 |
input = st.text_input("Query ...")
|
| 35 |
+
st.text(p.run(str(input or "test")))
|
| 36 |
+
|
| 37 |
|
| 38 |
if __name__ == "__main__":
|
| 39 |
+
main()
|
faiss_document_store.db
ADDED
|
Binary file (73.7 kB). View file
|
|
|
my_faiss_config.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
{"faiss_config_path": "my_faiss_config.json", "embedding_dim": 768}
|
my_faiss_index.faiss
ADDED
|
Binary file (6.19 kB). View file
|
|
|
requirements.txt
CHANGED
|
@@ -1 +1 @@
|
|
| 1 |
-
farm-haystack
|
|
|
|
| 1 |
+
farm-haystack[faiss]
|