new code
Browse files- 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,
|