hanzla commited on
Commit
77ab3f6
·
1 Parent(s): e92f652
Files changed (1) hide show
  1. src/pdfchatbot.py +1 -2
src/pdfchatbot.py CHANGED
@@ -5,7 +5,6 @@ import gradio as gr
5
  from PIL import Image
6
  from langchain.embeddings import HuggingFaceEmbeddings
7
  from langchain.vectorstores import Chroma
8
- from langchain.llms import HuggingFacePipeline
9
  from langchain.chains import ConversationalRetrievalChain
10
  from langchain.document_loaders import PyPDFLoader
11
  from langchain.prompts import PromptTemplate
@@ -87,7 +86,7 @@ class PDFChatBot:
87
  self.pipeline.tokenizer.convert_tokens_to_ids("<|eot_id|>")
88
  ]
89
  temp = 0.1
90
- outputs = pipeline(
91
  prompt,
92
  max_new_tokens=1024,
93
  eos_token_id=terminators,
 
5
  from PIL import Image
6
  from langchain.embeddings import HuggingFaceEmbeddings
7
  from langchain.vectorstores import Chroma
 
8
  from langchain.chains import ConversationalRetrievalChain
9
  from langchain.document_loaders import PyPDFLoader
10
  from langchain.prompts import PromptTemplate
 
86
  self.pipeline.tokenizer.convert_tokens_to_ids("<|eot_id|>")
87
  ]
88
  temp = 0.1
89
+ outputs = self.pipeline(
90
  prompt,
91
  max_new_tokens=1024,
92
  eos_token_id=terminators,