AstraBert commited on
Commit
9628227
1 Parent(s): 95b36ab
Files changed (1) hide show
  1. utils.py +1 -3
utils.py CHANGED
@@ -12,7 +12,7 @@ import os
12
  from pypdf import PdfMerger
13
 
14
 
15
- mod = "google/flan-t5-large"
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