Duplicate from johnnyfivefingers/summarymachine

Co-authored-by: JH <johnnyfivefingers@users.noreply.huggingface.co>

.gitattributes

@@ -0,0 +1,27 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zstandard filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,13 @@
+---
+title: Summarymachine
+emoji: 🌖
+colorFrom: pink
+colorTo: pink
+sdk: gradio
+sdk_version: 3.0.2
+app_file: app.py
+pinned: false
+duplicated_from: johnnyfivefingers/summarymachine
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces#reference

app.py

@@ -0,0 +1,31 @@
+import gradio as gr
+
+from transformers import pipeline
+import csv
+
+model_id = "philschmid/bart-large-cnn-samsum"
+
+summarizer = pipeline("summarization", 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)
+    
+    
+    generated_summary_short = summarizer(text, max_length=40, min_length=10)[0]['summary_text']
+    generated_summary = summarizer(text, max_length=80, min_length=20)[0]['summary_text']
+    generated_summary_long = summarizer(text, max_length=200, min_length=40)[0]['summary_text']
+    
+    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)
+
+iface = gr.Interface(fn=summarize, inputs="text", outputs="text")
+iface.launch()

requirements.txt

@@ -0,0 +1,3 @@
+transformers
+tensorflow
+torch