Update app.py
Browse filesCorrect pipeline mistake in the transformer import
app.py
CHANGED
|
@@ -12,13 +12,13 @@ from langchain_community.document_loaders import PyPDFLoader
|
|
| 12 |
from langchain_classic.chains import RetrievalQA
|
| 13 |
from langchain_community.llms import HuggingFacePipeline
|
| 14 |
from langchain_community.embeddings import HuggingFaceEmbeddings
|
| 15 |
-
from langchain_huggingface import HuggingFacePipeline
|
| 16 |
import torch
|
| 17 |
|
| 18 |
from huggingface_hub import HfFolder
|
| 19 |
import gradio as gr
|
| 20 |
|
| 21 |
-
from transformers import AutoTokenizer, AutoModelForCausalLM,
|
| 22 |
|
| 23 |
## You can use this section to suppress warnings generated by your code:
|
| 24 |
def warn(*args, **kwargs):
|
|
|
|
| 12 |
from langchain_classic.chains import RetrievalQA
|
| 13 |
from langchain_community.llms import HuggingFacePipeline
|
| 14 |
from langchain_community.embeddings import HuggingFaceEmbeddings
|
| 15 |
+
# from langchain_huggingface import HuggingFacePipeline
|
| 16 |
import torch
|
| 17 |
|
| 18 |
from huggingface_hub import HfFolder
|
| 19 |
import gradio as gr
|
| 20 |
|
| 21 |
+
from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline, BitsAndBytesConfig
|
| 22 |
|
| 23 |
## You can use this section to suppress warnings generated by your code:
|
| 24 |
def warn(*args, **kwargs):
|