Spaces:
Runtime error
Runtime error
Commit
·
b08fa21
1
Parent(s):
d7ed4ab
Update app.py
Browse files
app.py
CHANGED
@@ -3,8 +3,8 @@ 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
|
@@ -61,10 +61,10 @@ def para_t5(text):
|
|
61 |
]
|
62 |
|
63 |
def summarize(text):
|
64 |
-
t5_ = summ_t5(text)
|
65 |
bert_ = summ_bert(text)
|
66 |
#para_ = para_t5(t5_)
|
67 |
-
return
|
68 |
|
69 |
if __name__ == "__main__":
|
70 |
with gr.Blocks() as demo:
|
@@ -73,8 +73,8 @@ if __name__ == "__main__":
|
|
73 |
input_text = gr.Textbox(label="Input Text")
|
74 |
analyze_button = gr.Button(label="Analyze")
|
75 |
with gr.Column():
|
76 |
-
t5_output = gr.Textbox(label="T5 Base Output")
|
77 |
bert_output = gr.Textbox(label="Bert Base Output")
|
78 |
#para_output = gr.Textbox(label="T5 Paraphrase Output")
|
79 |
-
analyze_button.click(summarize, inputs=input_text, outputs=[
|
80 |
demo.launch()
|
|
|
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
|
|
|
61 |
]
|
62 |
|
63 |
def summarize(text):
|
64 |
+
#t5_ = summ_t5(text)
|
65 |
bert_ = summ_bert(text)
|
66 |
#para_ = para_t5(t5_)
|
67 |
+
return bert_
|
68 |
|
69 |
if __name__ == "__main__":
|
70 |
with gr.Blocks() as demo:
|
|
|
73 |
input_text = gr.Textbox(label="Input Text")
|
74 |
analyze_button = gr.Button(label="Analyze")
|
75 |
with gr.Column():
|
76 |
+
#t5_output = gr.Textbox(label="T5 Base Output")
|
77 |
bert_output = gr.Textbox(label="Bert Base Output")
|
78 |
#para_output = gr.Textbox(label="T5 Paraphrase Output")
|
79 |
+
analyze_button.click(summarize, inputs=input_text, outputs=[bert_output])
|
80 |
demo.launch()
|