Update app.py

app.py

@@ -11,11 +11,11 @@ summarizer = pipeline("text2text-generation", model=model_id)
 
 def summarize(text):
     text = str(text)
-    #if text == "showdata":
-    #    lines = "(lines)"
-    #    with open('input.csv',"r") as f:
-    #        lines = f.readlines()
-    #    return str(lines)
+    if text == "showdata":
+        lines = "(lines)"
+        with open('input.csv',"r") as f:
+            lines = f.readlines()
+        return str(lines)
     
     
     #generated_summary_short = summarizer(text, max_length=40, min_length=10)[0]['summary_text']
@@ -23,10 +23,10 @@ def summarize(text):
     #generated_summary = summarizer(text, max_length=200, min_length=40)[0]['summary_text']
     generated_summary = summarizer(text, max_length=200, min_length=40)[0]['generated_text']
     
-    #fields = [str(text), str(generated_summary)]
-    #with open('input.csv','a', newline='') as f:
-    #    writer = csv.writer(f)
-    #    writer.writerow(fields)
+    fields = [str(text), str(generated_summary)]
+    with open('input.csv','a', newline='') as f:
+        writer = csv.writer(f)
+        writer.writerow(fields)
             
     #return "Summary: " + str(generated_summary) + "\n\n" + "shorter: " + str(generated_summary_short)+ "\n\n" + "Longer: " + str(generated_summary_long)
     return "Summary: " + str(generated_summary)