Slider

app.py

@@ -19,25 +19,25 @@ examples = [
     Q: A juggler can juggle 16 balls. Half of the balls are golf balls, and half of the golf balls are blue. How many blue golf balls are there?"""]
 ]
 
-title = "Upword - Models Competition"
-description = "This demo compares [BART-Large-CNN](https://huggingface.co/facebook/bart-large-cnn) and [Flan-T5-XXL](https://huggingface.co/google/flan-t5-xxl)."
+title = "Upword. Model Playground"
 
 token = os.environ["token"]
 
 urls = {
     'flan-t5': os.environ["url"],
     # 'flan-t5': "https://api-inference.huggingface.co/models/philschmid/flan-t5-xxl-sharded-fp16",
-    'bart-large-cnn': "https://api-inference.huggingface.co/models/facebook/bart-large-cnn"
+    'bart-large-cnn': "https://api-inference.huggingface.co/models/facebook/bart-large-cnn",
+    'pegasus': "https://api-inference.huggingface.co/models/google/pegasus-xsum"
 }
 
 
-def inference(text):
+def inference(text, min_length, max_length):
     headers = {"Authorization": f"Bearer {token}"}
     payload = {
         "inputs": text,
         "parameters": {
-            "min_length": 30,
-            "max_length": 120,
+            "min_length": min_length,
+            "max_length": max_length,
             "do_sample": False
         }
     }
@@ -46,19 +46,24 @@ def inference(text):
         responses[model] = requests.post(url, headers=headers, json=payload)
 
     output_flan = responses['flan-t5'].json()[0]['generated_text']
-    output_vanilla = responses['bart-large-cnn'].json()[0]['summary_text']
-    return [output_flan, output_vanilla]
+    output_bart = responses['bart-large-cnn'].json()[0]['summary_text']
+    output_gpt = responses['pegasus'].json()[0]['summary_text']
+    return [output_flan, output_bart, output_gpt]
 
 
 io = gr.Interface(
   inference,
-  gr.Textbox(lines=3),
+  inputs=[
+      gr.Textbox(label='Input', lines=3),
+      gr.Slider(minimum=1, maximum=160, value=20, label="min_length"),
+      gr.Slider(minimum=1, maximum=160, value=80, label="max_length")
+  ],
   outputs=[
     gr.Textbox(lines=3, label="Flan T5-XXL"),
-    gr.Textbox(lines=3, label="BART-Large-CNN")
+    gr.Textbox(lines=3, label="BART-Large-CNN"),
+    gr.Textbox(lines=3, label="Pegasus")
   ],
   title=title,
-  description=description,
   examples=examples
 )
 io.launch()