Summarizaation app

Makefile

@@ -0,0 +1,14 @@
+install:
+	python -m pip install --upgrade pip &&\
+		python -m pip install -r requirements.txt
+
+test:
+	python -m pytest -vv test_app.py
+
+format:
+	black *.py
+
+lint:
+	python -m pylint --disable=R,C app.py
+
+all: install lint test format

app.py

@@ -0,0 +1,14 @@
+from transformers import pipeline
+import gradio as gr
+
+model = pipeline("summarization")
+
+def predict(prompt):
+    summary = model(prompt)[0]["summary_text"]
+    return summary
+
+with gr.Blocks() as demo:
+    textbox = gr.Textbox(placeholder="Enter text block to summarize", lines=4)
+    gr.Interface(fn=predict, inputs=textbox, outputs="text")
+
+demo.launch()

requirements.txt

@@ -0,0 +1,7 @@
+gradio
+transformers
+tensorflow
+pylint
+pytest
+black
+ipython