Update app.py

app.py

@@ -6,10 +6,14 @@ from llama_index.vector_stores.chroma import ChromaVectorStore
 from llama_index.core import StorageContext
 from llama_index.embeddings.huggingface import HuggingFaceEmbedding
 import torch
-from llama_index.core import VectorStoreIndex
-from llama_index.core import Document
-from llama_index.core import Settings 
-from llama_index.llms.huggingface import (HuggingFaceInferenceAPI, HuggingFaceLLM, )
+from llama_index.core import (
+    VectorStoreIndex,
+    Document,
+    Settings,
+    ChatMessage,
+)
+from llama_index.llms.huggingface import (HuggingFaceLLM, )
+from llama_index.llms.huggingface.api import (HuggingFaceInferenceAPI, )
 from huggingface_hub import login
 import chromadb as chromadb
 from chromadb.utils import embedding_functions