KalooAI commited on
Commit
1a96650
·
verified ·
1 Parent(s): d9ab9ba

Update app.py

Browse files

Correct pipeline mistake in the transformer import

Files changed (1) hide show
  1. app.py +2 -2
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, Pipeline, BitsAndBytesConfig
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):