alaselababatunde commited on
Commit
cc8bd1e
·
1 Parent(s): 15d6738
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -8,7 +8,7 @@ import torch
8
  import logging
9
 
10
  # LangChain imports (modern >=1.0)
11
- from langchain.chat_models import HuggingFacePipeline
12
  from langchain.chains import LLMChain
13
  from langchain.prompts.prompt import PromptTemplate
14
  from langchain.memory import ConversationBufferMemory
 
8
  import logging
9
 
10
  # LangChain imports (modern >=1.0)
11
+ from langchain.llms.huggingface_pipeline import HuggingFacePipeline
12
  from langchain.chains import LLMChain
13
  from langchain.prompts.prompt import PromptTemplate
14
  from langchain.memory import ConversationBufferMemory