update march14

app.py

@@ -0,0 +1,41 @@
+from transformers import pipeline
+import gradio as gr
+
+# the summarization pipeline
+sum_pipeline = pipeline(
+    "summarization",
+    model="kaiku03/Summarization_story"
+)
+
+# function for the gradio app
+def fn_summarization(prompt):
+    return sum_pipeline(prompt)[0]['summary_text']
+
+# gradio app
+# Define example inputs and outputs
+examples = ["""One day a rabbit was boasting about how fast he could run. 
+             He was laughing at the turtle for being so slow. 
+             Much to the rabbit’s surprise, the turtle challenged him to a race. 
+             The rabbit thought this was a good joke and accepted the challenge. 
+             The fox was to be the umpire of the race. As the race began, 
+             the rabbit raced way ahead of the turtle, just like everyone thought.
+             The rabbit got to the halfway point and could not see the turtle anywhere. 
+             He was hot and tired and decided to stop and take a short nap.
+             Even if the turtle passed him, he would be able to race to the finish line ahead of him. 
+             All this time the turtle kept walking step by step by step. He never quit no matter how hot or tired he got.
+             He just kept going. However, the rabbit slept longer than he had thought and woke up.
+             He could not see the turtle anywhere! He went at full speed to the finish line but found the turtle there waiting for him."""]
+
+iface = gr.Interface(
+    fn=fn_summarization,
+    inputs='text',
+    outputs=gr.Textbox(label="Summary"),
+    examples=[
+        [ex] for ex in examples
+    ],
+    title='Text Summarization',
+    description='Summarize your text using Kaiku03/Summarization_story model.',
+    article='All done by Kaiku'
+)
+
+iface.launch()

requirements.txt

@@ -0,0 +1,5 @@
+transformers
+gradio==3.45
+torch
+sentencepiece
+nltk