Hugging Face's logo

Spaces:
nikhmr1235
/
PDF_document_chatbot
Sleeping

App Files Community
nikhmr1235 commited on
Commit
124f540
·
verified ·
1 Parent(s): c89d558

swicth from BAAI/bge-large-en-v1.5 to google generative model

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -19,7 +19,8 @@ nest_asyncio.apply()
19
 
20
  # Constants
21
  LLM_MODEL = "gemini-1.5-flash"
22
- EMBEDDING_MODEL = "BAAI/bge-large-en-v1.5"
 
23
  CHROMA_DB_PATH = tempfile.gettempdir() + "/chroma_db"
24
 
25
  class PDFChatbot:
 
19
 
20
  # Constants
21
  LLM_MODEL = "gemini-1.5-flash"
22
+
23
+ EMBEDDING_MODEL = "models/embedding-001"
24
  CHROMA_DB_PATH = tempfile.gettempdir() + "/chroma_db"
25
 
26
  class PDFChatbot: