Eitan Lifshits commited on
Commit
46d7acf
·
1 Parent(s): fcd9f1e

fix output format

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -7,13 +7,11 @@ summarizer = pipeline('summarization', model='Eitanli/resume_label_summary_model
7
 
8
  def predict(text):
9
  summary = summarizer(text, max_length=10, min_length=2)
10
- return {'summary': summary['summary_text']}
11
 
12
 
13
  iface = gr.Interface(
14
  fn=predict,
15
  inputs='text',
16
  outputs="text")
17
-
18
- if __name__ == "__main__":
19
- iface.launch()
 
7
 
8
  def predict(text):
9
  summary = summarizer(text, max_length=10, min_length=2)
10
+ return {'summary': summary}
11
 
12
 
13
  iface = gr.Interface(
14
  fn=predict,
15
  inputs='text',
16
  outputs="text")
17
+ iface.launch()