Update src/pdfchatbot.py
Browse files- src/pdfchatbot.py +1 -1
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):
|
|
|
|
| 60 |
print("Vector store created")
|
| 61 |
@spaces.GPU
|
| 62 |
def load_tokenizer(self):
|
| 63 |
+
self.tokenizer = AutoTokenizer.from_pretrained("kiddothe2b/hierarchical-transformer-base-4096")
|
| 64 |
|
| 65 |
@spaces.GPU
|
| 66 |
def create_organic_pipeline(self):
|