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,
|