wldmr commited on
Commit
e76946d
1 Parent(s): c9c32ea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -21,7 +21,7 @@ def summarize(text):
21
  #generated_summary_short = summarizer(text, max_length=40, min_length=10)[0]['summary_text']
22
  #generated_summary = summarizer(text, max_length=80, min_length=20)[0]['summary_text']
23
  #generated_summary = summarizer(text, max_length=200, min_length=40)[0]['summary_text']
24
- generated_summary = summarizer(text, max_length=200, min_length=40)[0]
25
 
26
  fields = [str(text), str(generated_summary)]
27
  with open('input.csv','a', newline='') as f:
@@ -29,7 +29,7 @@ def summarize(text):
29
  writer.writerow(fields)
30
 
31
  #return "Summary: " + str(generated_summary) + "\n\n" + "shorter: " + str(generated_summary_short)+ "\n\n" + "Longer: " + str(generated_summary_long)
32
- return "Summary: " + str(generated_summary)
33
 
34
 
35
  iface = gr.Interface(fn=summarize, inputs="text", outputs="text")
 
21
  #generated_summary_short = summarizer(text, max_length=40, min_length=10)[0]['summary_text']
22
  #generated_summary = summarizer(text, max_length=80, min_length=20)[0]['summary_text']
23
  #generated_summary = summarizer(text, max_length=200, min_length=40)[0]['summary_text']
24
+ generated_summary = summarizer(text, max_length=200, min_length=40)[0]['generated_text']
25
 
26
  fields = [str(text), str(generated_summary)]
27
  with open('input.csv','a', newline='') as f:
 
29
  writer.writerow(fields)
30
 
31
  #return "Summary: " + str(generated_summary) + "\n\n" + "shorter: " + str(generated_summary_short)+ "\n\n" + "Longer: " + str(generated_summary_long)
32
+ return str(generated_summary)
33
 
34
 
35
  iface = gr.Interface(fn=summarize, inputs="text", outputs="text")