Update app.py
Browse files
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)
|