summarization test

Makefile

@@ -0,0 +1,23 @@
+install:
+	pip install --upgrade pip &&\
+		pip install -r requirements.txt
+
+test:
+	python -m pytest -vv test_main.py
+
+format:
+	black *.py
+
+run:
+	python app.py
+
+run-uvicorn:
+	uvicorn main:app --reload
+
+killweb:
+	sudo killall uvicorn
+
+lint:
+	pylint --disable=R,C main.py
+
+all: install lint

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,3 @@
+gradio 
+transformers
+torch