wldmr commited on
Commit
e83b9e0
·
1 Parent(s): abd582a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -16,8 +16,8 @@ def summarize(text):
16
  return str(lines)
17
 
18
 
19
- generated_summary_short = summarizer(text, max_length=40, min_length=10)[0]['summary_text']
20
- generated_summary = summarizer(text, max_length=80, min_length=20)[0]['summary_text']
21
  generated_summary_long = summarizer(text, max_length=200, min_length=40)[0]['summary_text']
22
 
23
  fields = [str(text), str(generated_summary)]
@@ -25,7 +25,9 @@ def summarize(text):
25
  writer = csv.writer(f)
26
  writer.writerow(fields)
27
 
28
- return "Summary: " + str(generated_summary) + "\n\n" + "shorter: " + str(generated_summary_short)+ "\n\n" + "Longer: " + str(generated_summary_long)
 
 
29
 
30
  iface = gr.Interface(fn=summarize, inputs="text", outputs="text")
31
  iface.launch()
 
16
  return str(lines)
17
 
18
 
19
+ #generated_summary_short = summarizer(text, max_length=40, min_length=10)[0]['summary_text']
20
+ #generated_summary = summarizer(text, max_length=80, min_length=20)[0]['summary_text']
21
  generated_summary_long = summarizer(text, max_length=200, min_length=40)[0]['summary_text']
22
 
23
  fields = [str(text), str(generated_summary)]
 
25
  writer = csv.writer(f)
26
  writer.writerow(fields)
27
 
28
+ #return "Summary: " + str(generated_summary) + "\n\n" + "shorter: " + str(generated_summary_short)+ "\n\n" + "Longer: " + str(generated_summary_long)
29
+ return "Summary: " + str(generated_summary_long)
30
+
31
 
32
  iface = gr.Interface(fn=summarize, inputs="text", outputs="text")
33
  iface.launch()