Najish Mahmud commited on
Commit
0772a3b
·
1 Parent(s): 1654d64

fixed models name

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from normalizer import normalize
7
  def summarize(text):
8
  model = AutoModelForSeq2SeqLM.from_pretrained("models/Afsara/csebuetnlp-banglat5")
9
  tokenizer = AutoTokenizer.from_pretrained(
10
- "models/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)
 
7
  def summarize(text):
8
  model = AutoModelForSeq2SeqLM.from_pretrained("models/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)