Spaces:
Runtime error
Runtime error
Commit
·
b6adfce
1
Parent(s):
aee9fe8
Update app.py
Browse files
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 |
-
|
7 |
-
|
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 |
-
|
15 |
-
|
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):
|