johnnyfivefingers commited on
Commit
05ace25
·
1 Parent(s): 0dd3e75

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -20,14 +20,19 @@ summarizer = pipeline("summarization", model=model_id)
20
  def summarize(text):
21
  #return "Summary: " + text
22
  text = str(text)
23
- generated_summary = summarizer(text, max_length=150, min_length = 20)
24
 
25
  fields = [text, generated_summary]
26
  with open('input.csv','a', newline='') as f:
27
  writer = csv.writer(f)
28
  writer.writerow(fields)
29
 
30
- return "Summary: " + str(generated_summary)
 
 
 
 
 
31
 
32
  iface = gr.Interface(fn=summarize, inputs="text", outputs="text")
33
  iface.launch()
 
20
  def summarize(text):
21
  #return "Summary: " + text
22
  text = str(text)
23
+ generated_summary = summarizer(text, max_length=150, min_length = 10)
24
 
25
  fields = [text, generated_summary]
26
  with open('input.csv','a', newline='') as f:
27
  writer = csv.writer(f)
28
  writer.writerow(fields)
29
 
30
+ lines = "(lines)"
31
+ f = open('input.csv',"r")
32
+ lines = f.readlines()
33
+ #print(lines)
34
+
35
+ return "Summary: " + str(generated_summary) + "\n\n" + str(lines)
36
 
37
  iface = gr.Interface(fn=summarize, inputs="text", outputs="text")
38
  iface.launch()