wiraindrak commited on
Commit
b6adfce
·
1 Parent(s): aee9fe8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -3,16 +3,16 @@ from transformers import T5Tokenizer, T5Model, T5ForConditionalGeneration, BertT
3
  import gradio as gr
4
  from gradio.mix import Parallel
5
 
6
- #tokenizer_t5 = T5Tokenizer.from_pretrained("panggi/t5-base-indonesian-summarization-cased")
7
- #model_t5 = T5ForConditionalGeneration.from_pretrained("panggi/t5-base-indonesian-summarization-cased")
8
 
9
  tokenizer_bert = BertTokenizer.from_pretrained("cahya/bert2bert-indonesian-summarization")
10
  tokenizer_bert.bos_token = tokenizer_bert.cls_token
11
  tokenizer_bert.eos_token = tokenizer_bert.sep_token
12
  model_bert = EncoderDecoderModel.from_pretrained("cahya/bert2bert-indonesian-summarization")
13
 
14
- #t5_para_tokenizer = AutoTokenizer.from_pretrained("Wikidepia/IndoT5-base-paraphrase")
15
- #t5_para_model = AutoModelForSeq2SeqLM.from_pretrained("Wikidepia/IndoT5-base-paraphrase")
16
 
17
 
18
  def summ_t5(text):
 
3
  import gradio as gr
4
  from gradio.mix import Parallel
5
 
6
+ tokenizer_t5 = T5Tokenizer.from_pretrained("panggi/t5-base-indonesian-summarization-cased")
7
+ model_t5 = T5ForConditionalGeneration.from_pretrained("panggi/t5-base-indonesian-summarization-cased")
8
 
9
  tokenizer_bert = BertTokenizer.from_pretrained("cahya/bert2bert-indonesian-summarization")
10
  tokenizer_bert.bos_token = tokenizer_bert.cls_token
11
  tokenizer_bert.eos_token = tokenizer_bert.sep_token
12
  model_bert = EncoderDecoderModel.from_pretrained("cahya/bert2bert-indonesian-summarization")
13
 
14
+ t5_para_tokenizer = AutoTokenizer.from_pretrained("Wikidepia/IndoT5-base-paraphrase")
15
+ t5_para_model = AutoModelForSeq2SeqLM.from_pretrained("Wikidepia/IndoT5-base-paraphrase")
16
 
17
 
18
  def summ_t5(text):