Update app.py

app.py

@@ -5,8 +5,7 @@ from langchain.prompts import ChatPromptTemplate
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain_huggingface import HuggingFaceEmbeddings
 from transformers import pipeline
-from langchain import HuggingFacePipeline
-
+from langchain_community import llms.huggingface_pipeline.HuggingFacePipeline
 from langchain.retrievers import ParentDocumentRetriever
 from langchain.storage import InMemoryStore
 from langchain_chroma import Chroma
@@ -83,18 +82,12 @@ def get_chain(temperature):
     # Replace the local OLMOLLM with the Hugging Face model
     
     model_name = "gpt2"
-    pipe = pipeline("text-generation", model=model_name)
+    pipe = pipeline("text-generation", model=model_name, max_length=100, max_new_tokens=50, temperature = temperature)
+
     
     # Initialize the LangChain HuggingFacePipeline
-    llm = HuggingFacePipeline.from_model_id(
-    model_id="microsoft/Phi-3-mini-4k-instruct",
-    task="text-generation",
-    pipeline_kwargs={
-        "max_new_tokens": 100,
-        "top_k": 50,
-        "temperature": temperature,
-    },
-)
+    llm = HuggingFacePipeline(pipeline=pipe)
+
     
     # Initialize LangChain
     # llm = HuggingFaceLLM(