Najish Mahmud commited on
Commit
434c95e
·
1 Parent(s): ccf2739

added token

Browse files
Files changed (2) hide show
  1. .gitignore +1 -0
  2. app.py +3 -2
.gitignore CHANGED
@@ -1 +1,2 @@
1
  .venv
 
 
1
  .venv
2
+ predict.py
app.py CHANGED
@@ -5,9 +5,10 @@ from normalizer import normalize
5
 
6
 
7
  def summarize(text):
8
- model = AutoModelForSeq2SeqLM.from_pretrained("Afsara/csebuetnlp-banglat5")
 
9
  tokenizer = AutoTokenizer.from_pretrained(
10
- "Afsara/csebuetnlp-banglat5", use_fast=False)
11
 
12
  input_ids = tokenizer(normalize(text), return_tensors="pt").input_ids
13
  generated_tokens = model.generate(input_ids)
 
5
 
6
 
7
  def summarize(text):
8
+ model = AutoModelForSeq2SeqLM.from_pretrained(
9
+ "Afsara/csebuetnlp-banglat5", use_auth_token="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")
10
  tokenizer = AutoTokenizer.from_pretrained(
11
+ "Afsara/csebuetnlp-banglat5", use_auth_token="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", use_fast=False)
12
 
13
  input_ids = tokenizer(normalize(text), return_tensors="pt").input_ids
14
  generated_tokens = model.generate(input_ids)