Updated

main.py

@@ -1,13 +1,14 @@
 from fastapi import FastAPI, HTTPException, Header
 from pydantic import BaseModel
 from transformers import pipeline
-from langchain.llms import HuggingFacePipeline
+from langchain import HuggingFacePipeline
 from langchain.chains import LLMChain
 from langchain.prompts import PromptTemplate
 from langchain.memory import ConversationBufferMemory
 import torch
 import logging
 
+
 # ===============================================
 # CONFIGURATION
 # ===============================================