Update initialize.py

initialize.py

@@ -1,8 +1,9 @@
-from haystack import Document, Secret
+from haystack import Document
+from haystack.utils import Secret
 from haystack.document_stores.in_memory import InMemoryDocumentStore
 from haystack.components.retrievers.in_memory import InMemoryBM25Retriever
 from haystack.components.builders import PromptBuilder
-from haystack.components.generators.hugging_face_local import HuggingFaceLocalGenerator
+from haystack.components.generators import HuggingFaceTGIGenerator
 from haystack.pipeline import Pipeline
 
 def init_doc_store(path, files):
@@ -33,7 +34,7 @@ def define_components(document_store, api_key):
     prompt_builder = PromptBuilder(template=template)
     
     generator = HuggingFaceTGIGenerator(
-        model="mistralai/Mistral-7B-v0.1", 
+        model="mistralai/Mistral-7B-Instruct-v0.1", 
         token=Secret.from_token(api_key)
     )
     generator.warm_up()