Add summarizer

app.py

@@ -1,7 +1,10 @@
 import gradio as gr
+from predict import summarize
 
-def greet(name):
-    return "Hello " + name + "!!"
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+def summarize_text(text):
+    return summarize(text)
+
+
+iface = gr.Interface(fn=summarize_text, inputs="text", outputs="text")
+iface.launch()

predict.py

@@ -0,0 +1,8 @@
+from summarizer import Summarizer
+from transformers import pipeline
+
+
+def summarize(x):
+    summarizer = Summarizer(model='distilbert-base-uncased')
+    output = summarizer(x, ratio=0.2, return_as_list=False)
+    return output