Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
|
2 |
+
import gradio as gr
|
|
|
|
|
|
|
|
|
|
|
|
|
3 |
|
4 |
+
gr.Interface.load("models/Afsara/fb_bart_large_cnn").launch()
|