wiraindrak commited on
Commit
b08fa21
·
1 Parent(s): d7ed4ab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
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 t5_, bert_
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=[t5_output, bert_output])
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()