Update app.py
Browse files
app.py
CHANGED
@@ -18,7 +18,7 @@ load_dotenv()
|
|
18 |
GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
|
19 |
|
20 |
embed_model = HuggingFaceEmbedding(model_name="BAAI/bge-small-en")
|
21 |
-
loader = SimpleDirectoryReader(input_dir="
|
22 |
documents = loader.load_data()
|
23 |
llm = Gemini(api_key=GOOGLE_API_KEY)
|
24 |
db = chromadb.PersistentClient(path="./chroma_db")
|
|
|
18 |
GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
|
19 |
|
20 |
embed_model = HuggingFaceEmbedding(model_name="BAAI/bge-small-en")
|
21 |
+
loader = SimpleDirectoryReader(input_dir=".", required_exts=[".pdf"])
|
22 |
documents = loader.load_data()
|
23 |
llm = Gemini(api_key=GOOGLE_API_KEY)
|
24 |
db = chromadb.PersistentClient(path="./chroma_db")
|