alaselababatunde commited on
Commit
83532f7
·
1 Parent(s): 02b3d0d
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -4,7 +4,7 @@ import torch
4
  import logging
5
 
6
  # LangChain imports (stable with latest version >=1.0)
7
- from langchain.llms import HuggingFacePipeline
8
  from langchain.chains import LLMChain
9
  from langchain.prompts import PromptTemplate
10
  from langchain.memory import ConversationBufferMemory
 
4
  import logging
5
 
6
  # LangChain imports (stable with latest version >=1.0)
7
+ from langchain.llms.huggingface_pipeline import HuggingFacePipeline
8
  from langchain.chains import LLMChain
9
  from langchain.prompts import PromptTemplate
10
  from langchain.memory import ConversationBufferMemory