Update src/pdfchatbot.py
Browse files- src/pdfchatbot.py +2 -2
src/pdfchatbot.py
CHANGED
|
@@ -60,7 +60,7 @@ class PDFChatBot:
|
|
| 60 |
print("Vector store created")
|
| 61 |
@spaces.GPU
|
| 62 |
def load_tokenizer(self):
|
| 63 |
-
self.tokenizer = AutoTokenizer.from_pretrained("
|
| 64 |
|
| 65 |
@spaces.GPU
|
| 66 |
def create_organic_pipeline(self):
|
|
@@ -102,7 +102,7 @@ class PDFChatBot:
|
|
| 102 |
temp = 0.1
|
| 103 |
outputs = self.pipe(
|
| 104 |
prompt,
|
| 105 |
-
max_new_tokens=
|
| 106 |
do_sample=True,
|
| 107 |
temperature=temp,
|
| 108 |
top_p=0.9,
|
|
|
|
| 60 |
print("Vector store created")
|
| 61 |
@spaces.GPU
|
| 62 |
def load_tokenizer(self):
|
| 63 |
+
self.tokenizer = AutoTokenizer.from_pretrained("microsoft/Phi-3-mini-128k-instruct")
|
| 64 |
|
| 65 |
@spaces.GPU
|
| 66 |
def create_organic_pipeline(self):
|
|
|
|
| 102 |
temp = 0.1
|
| 103 |
outputs = self.pipe(
|
| 104 |
prompt,
|
| 105 |
+
max_new_tokens=2048,
|
| 106 |
do_sample=True,
|
| 107 |
temperature=temp,
|
| 108 |
top_p=0.9,
|