Najish Mahmud commited on
Commit
c2a01c8
·
1 Parent(s): 30e8cb9

fixed input output

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -5,9 +5,9 @@ from normalizer import normalize
5
 
6
 
7
  def summarize(text):
8
- model = AutoModelForSeq2SeqLM.from_pretrained("csebuetnlp/banglat5")
9
  tokenizer = AutoTokenizer.from_pretrained(
10
- "csebuetnlp/banglat5", use_fast=False)
11
 
12
  input_ids = tokenizer(normalize(text), return_tensors="pt").input_ids
13
  generated_tokens = model.generate(input_ids)
@@ -15,6 +15,6 @@ def summarize(text):
15
  return decoded_tokens
16
 
17
 
18
- gr_interface = gr.Interface("models/Afsara/fb_bart_large_cnn")
19
 
20
  gr_interface.launch()
 
5
 
6
 
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)
 
15
  return decoded_tokens
16
 
17
 
18
+ gr_interface = gr.Interface(fn=summarize, inputs="text", outputs="text")
19
 
20
  gr_interface.launch()