isayahc commited on
Commit
7961699
·
1 Parent(s): 6e37066

added transformer pipeline

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -8,7 +8,7 @@ from botocore.client import Config
8
 
9
  from huggingface_hub import AsyncInferenceClient
10
 
11
- from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
12
 
13
  from langchain.text_splitter import RecursiveCharacterTextSplitter
14
  from langchain.llms import HuggingFaceHub
@@ -17,6 +17,9 @@ from langchain.vectorstores import Chroma
17
  from langchain.chains import RetrievalQA
18
  from langchain.prompts import ChatPromptTemplate
19
  from langchain.document_loaders import WebBaseLoader
 
 
 
20
 
21
  from typing import Iterator
22
 
 
8
 
9
  from huggingface_hub import AsyncInferenceClient
10
 
11
+ from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer, pipeline
12
 
13
  from langchain.text_splitter import RecursiveCharacterTextSplitter
14
  from langchain.llms import HuggingFaceHub
 
17
  from langchain.chains import RetrievalQA
18
  from langchain.prompts import ChatPromptTemplate
19
  from langchain.document_loaders import WebBaseLoader
20
+ from langchain.llms.huggingface_pipeline import HuggingFacePipeline
21
+
22
+
23
 
24
  from typing import Iterator
25