File size: 1,043 Bytes
fbfd6f7
593b58b
 
fbfd6f7
593b58b
 
 
010f328
593b58b
fbfd6f7
 
 
593b58b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
fbfd6f7
 
593b58b
fbfd6f7
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
import gradio as gr
import re
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM

WHITESPACE_HANDLER = lambda k: re.sub('\s+', ' ', re.sub('\n+', ' ', k.strip()))

model_name = "csebuetnlp/mT5_multilingual_XLSum"
tokenizer = AutoTokenizer.from_pretrained(model_name,use_fast=False)
model = AutoModelForSeq2SeqLM.from_pretrained(model_name)

def generate_summary(text):

   input_ids = tokenizer(
    [WHITESPACE_HANDLER(text)],
    return_tensors="pt",
    padding="max_length",
    truncation=True,
    max_length=512)["input_ids"]

   output_ids = model.generate(
       input_ids=input_ids,
       max_length=84,
       no_repeat_ngram_size=2,
       num_beams=4
   )[0]

   summary = tokenizer.decode(
       output_ids,
       skip_special_tokens=True,
       clean_up_tokenization_spaces=False
   )

   return summary

demo = gr.Interface(fn=generate_summary,
                    inputs=gr.Textbox(lines=10, placeholder="Matinni kiriting!"),
                    outputs=gr.Textbox(lines=4)
                    )

demo.launch()