brezende commited on
Commit
3d587b0
1 Parent(s): 01ddaf6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- pipe = pipeline("summarization", model="lidiya/bart-large-xsum-samsum")
5
 
6
  def main(in_text):
7
  print(in_text)
@@ -22,7 +22,7 @@ Kleber: De acordo com minha pesquisa o número é 35 9 9898-6142"""
22
  output = gr.Textbox(label="Resultado")
23
  b1 = gr.Button("Summarize!")
24
  b1.click(main, inputs=[text1], outputs=output)
25
- gr.Markdown("""###### designed by [cloudqi]""")
26
 
27
 
28
  if __name__ == "__main__":
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ pipe = pipeline("summarization", model="cloudqi/cqi_brain_memory_summarizer_large_pt_v0")
5
 
6
  def main(in_text):
7
  print(in_text)
 
22
  output = gr.Textbox(label="Resultado")
23
  b1 = gr.Button("Summarize!")
24
  b1.click(main, inputs=[text1], outputs=output)
25
+ gr.Markdown("""###### Model cqi_brain_memory_summarizer_large_pt_v0 designed by [cloudqi]""")
26
 
27
 
28
  if __name__ == "__main__":