Spaces:

Jkalonji
/

DataChat

Runtime error

Jkalonji commited on Mar 6

Commit

9b8afb3

•

1 Parent(s): f816e10

point towards data folder

Files changed (1) hide show

app.py CHANGED Viewed

@@ -28,7 +28,7 @@ def RAG_Langchain(query):
     embeddings = SentenceTransformerEmbeddings(model_name=model_name)
     repo_id = "llmware/bling-sheared-llama-1.3b-0.1"
-    loader = DirectoryLoader('', glob="**/*.txt", show_progress=True, loader_cls=PyPDFLoader)
     documents = loader.load()

     embeddings = SentenceTransformerEmbeddings(model_name=model_name)
     repo_id = "llmware/bling-sheared-llama-1.3b-0.1"
+    loader = DirectoryLoader('data/', glob="**/*.pdf", show_progress=True, loader_cls=PyPDFLoader)
     documents = loader.load()