Afsara commited on
Commit
e1b2e4e
1 Parent(s): 6d63118

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -16
app.py CHANGED
@@ -1,18 +1,4 @@
1
- import gradio as gr
2
-
3
- from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
4
- from normalizer import normalize # pip install git+https://github.com/csebuetnlp/normalizer
5
-
6
- def summarize(text):
7
- model = AutoModelForSeq2SeqLM.from_pretrained("csebuetnlp/banglat5")
8
- tokenizer = AutoTokenizer.from_pretrained("csebuetnlp/banglat5", use_fast=False)
9
 
10
- input_ids = tokenizer(normalize(text), return_tensors="pt").input_ids
11
- generated_tokens = model.generate(input_ids)
12
- decoded_tokens = tokenizer.batch_decode(generated_tokens)[0]
13
- return decoded_tokens
14
-
15
-
16
- gr_interface = gr.Interface("models/Afsara/fb_bart_large_cnn")
17
 
18
- gr_interface.launch()
 
 
 
 
 
 
 
 
 
1
 
2
+ import gradio as gr
 
 
 
 
 
 
3
 
4
+ gr.Interface.load("models/Afsara/fb_bart_large_cnn").launch()