nightfury commited on
Commit
426a78e
·
verified ·
1 Parent(s): 7418ff3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -99,10 +99,12 @@ if __name__ == '__main__':
99
  """
100
  import chromadb
101
  from llama_index.vector_stores.chroma import ChromaVectorStore
102
- from llama_index.core import VectorStoreIndex, StorageContext, TextNode
 
103
  from llama_index.core.indices.vector_store.retrievers import VectorIndexAutoRetriever
104
  from llama_index.core.query_engine import RetrieverQueryEngine
105
  from llama_index.embeddings.huggingface import HuggingFaceEmbedding
 
106
 
107
  # Initialize ChromaDB client and collection
108
  chroma_client = chromadb.HttpClient(host="localhost", port="8080", ssl=False)
 
99
  """
100
  import chromadb
101
  from llama_index.vector_stores.chroma import ChromaVectorStore
102
+ #from llama_index.core import VectorStoreIndex, StorageContext, TextNode
103
+ from llama_index.core import VectorStoreIndex, SimpleDirectoryReader, ServiceContext, PromptTemplate
104
  from llama_index.core.indices.vector_store.retrievers import VectorIndexAutoRetriever
105
  from llama_index.core.query_engine import RetrieverQueryEngine
106
  from llama_index.embeddings.huggingface import HuggingFaceEmbedding
107
+ #from llama_index.llms.huggingface import HuggingFaceLLM
108
 
109
  # Initialize ChromaDB client and collection
110
  chroma_client = chromadb.HttpClient(host="localhost", port="8080", ssl=False)