Update model_utils.py

model_utils.py

@@ -19,8 +19,6 @@
 
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 from langchain.llms import HuggingFacePipeline
-from langchain.prompts import PromptTemplate
-from langchain.schemas import LangChainConfig
 
 
 MODEL_NAME = "microsoft/Phi-3-mini-4k-instruct"
@@ -76,12 +74,10 @@ def generate_text(hf_pipeline, input_text):
         input_variables=["input_text"],
         template="Please summarize the following text: {input_text}"
     )
-    # Configuration for LangChain (adjust max_tokens, temperature, etc., as needed)
-    config = LangChainConfig(max_tokens=50, temperature=0.7)
 
     # Example input text for the task
     # input_text = "LangChain is a library that facilitates the development of applications using language models."
     
     # Run the LangChain pipeline
-    output = hf_pipeline.generate(input_text, prompt_template=prompt_template, config=config)
+    output = hf_pipeline.generate(input_text, prompt_template=prompt_template)
     return output