Spaces:
Runtime error
Runtime error
AstraBert
commited on
Commit
·
9628227
1
Parent(s):
95b36ab
bugfix2
Browse files
utils.py
CHANGED
@@ -12,7 +12,7 @@ import os
|
|
12 |
from pypdf import PdfMerger
|
13 |
|
14 |
|
15 |
-
mod = "google/flan-t5-
|
16 |
tsk = "text2text-generation"
|
17 |
|
18 |
def merge_pdfs(pdfs: list):
|
@@ -131,14 +131,12 @@ def just_chatting(
|
|
131 |
try:
|
132 |
tokenizer = AutoTokenizer.from_pretrained(
|
133 |
mod,
|
134 |
-
trust_remote_code=True,
|
135 |
)
|
136 |
|
137 |
tokenizer.pad_token = tokenizer.eos_token
|
138 |
|
139 |
model = AutoModelForSeq2SeqLM.from_pretrained(
|
140 |
mod,
|
141 |
-
trust_remote_code=True,
|
142 |
)
|
143 |
except Exception as e:
|
144 |
import sys
|
|
|
12 |
from pypdf import PdfMerger
|
13 |
|
14 |
|
15 |
+
mod = "google/flan-t5-base"
|
16 |
tsk = "text2text-generation"
|
17 |
|
18 |
def merge_pdfs(pdfs: list):
|
|
|
131 |
try:
|
132 |
tokenizer = AutoTokenizer.from_pretrained(
|
133 |
mod,
|
|
|
134 |
)
|
135 |
|
136 |
tokenizer.pad_token = tokenizer.eos_token
|
137 |
|
138 |
model = AutoModelForSeq2SeqLM.from_pretrained(
|
139 |
mod,
|
|
|
140 |
)
|
141 |
except Exception as e:
|
142 |
import sys
|