Najish Mahmud commited on
Commit
dc0d991
·
1 Parent(s): fe4df75

removed token reqs

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,9 +6,9 @@ from normalizer import normalize
6
 
7
  def summarize(text):
8
  model = AutoModelForSeq2SeqLM.from_pretrained(
9
- "Afsara/cse_buet_bangla_t5", use_auth_token=True)
10
  tokenizer = AutoTokenizer.from_pretrained(
11
- "Afsara/cse_buet_bangla_t5", use_auth_token=True, use_fast=False)
12
 
13
  input_ids = tokenizer(normalize(text), return_tensors="pt").input_ids
14
  generated_tokens = model.generate(input_ids)
 
6
 
7
  def summarize(text):
8
  model = AutoModelForSeq2SeqLM.from_pretrained(
9
+ "Afsara/cse_buet_bangla_t5")
10
  tokenizer = AutoTokenizer.from_pretrained(
11
+ "Afsara/cse_buet_bangla_t5", use_fast=False)
12
 
13
  input_ids = tokenizer(normalize(text), return_tensors="pt").input_ids
14
  generated_tokens = model.generate(input_ids)