Two modes supported: "shorten", "paraphrase"

README.md

@@ -11,8 +11,10 @@ pinned: false
 license: openrail
 models:
   - facebook/bart-large-cnn
+  - Vamsi/T5_Paraphrase_Paws
 ---
 
-# Text summarizer
+# Text shortener/paraphraser
 
-Shortening texts using `facebook/bart-large-cnn`
+- Shortening texts using `facebook/bart-large-cnn`
+- Paraphrasing texts using `Vamsi/T5_Paraphrase_Paws`

app.py

@@ -29,16 +29,24 @@ def paraphrase_text(text, min_length, max_length):
   print("** outputs", len(outputs), line)
   return line
 
+def modify_text(mode, text, min_length, max_length):
+  if mode == "shorten":
+    return shorten_text(text, min_length, max_length)
+  else:
+    return paraphrase_text(text, min_length, max_length)
+
 gradio_interface = gradio.Interface(
-  fn=paraphrase_text,
+  fn=modify_text,
   inputs=[
+    gradio.Radio(["shorten", "paraphrase"], label="Mode"),
     "text",
     gradio.Slider(5, 200, value=30, label="Min length"),
     gradio.Slider(5, 500, value=130, label="Max length")
   ],
   outputs="text",
   examples=[
-    ["""A beautiful golden sun is setting. The sky is on fire. A large neon sign rises into shot. It rests on top of a skyscraper and fills the frame. The building is neither past nor future in design but a bit of both.
+    ["shorten",
+    """A beautiful golden sun is setting. The sky is on fire. A large neon sign rises into shot. It rests on top of a skyscraper and fills the frame. The building is neither past nor future in design but a bit of both.
 
 Slowly we pan downwards revealing the city that spreads below. A glittering conglomeration of elevated transport tubes, smaller square buildings which are merely huge, with, here and there, the comparatively minuscule relics of previous ages of architecture, pavement level awnings suggesting restaurants and shops. Transparent tubes carry whizzing transport cages past us. An elevated highway carrying traffic composed primarily of large transport lorries passes through frame.
 
@@ -46,8 +54,8 @@ As we descend, the sunlight is blocked out and street lights & neon signs take o
 Xmas decorations are everywhere. People are busy buying, ogling, discussing, choosing wisely from the goodies on display. Shoppers are going by laden with superbly packaged goods. The shop windows are full of elaborately boxed and be-ribboned who-knows-what. In one window is a bank of TV sets on the great majority of the screens is the face of Mr. Helpmann the Deputy Minister of Information. He is being interviewed. No-one bothers to listen to Helpmann.""",
     30, 130]
   ],
-  title="Text summarizer",
-  description="Shortening texts using `facebook/bart-large-cnn`.",
+  title="Text shortener/paraphraser",
+  description="Shortening texts using `facebook/bart-large-cnn`, paraphrasing texts using `Vamsi/T5_Paraphrase_Paws`.",
   article="© Tom Söderlund 2022"
 )
 gradio_interface.launch()