Added a slider for input length

app.py

@@ -13,11 +13,15 @@ summarizer = pipeline("summarization", model=MODEL_NAME, tokenizer=MODEL_NAME)
 # tokenizer = AutoTokenizer.from_pretrained(PRETRAIN_MODEL_NAME)
 
 # Define the function that will be used by the web app
-def summarize_dialog(dialog):
+def summarize_dialog(dialog, max_length):
     # Summarize the dialog
-    summary = summarizer(dialog, max_length=50, min_length=10, do_sample=False)
+    summary = summarizer(dialog, max_length=max_length, min_length=10, do_sample=False)
     # Return the summary
     return summary[0]['summary_text']
 
 # Define the web app & launch
-gr.Interface(fn=summarize_dialog, inputs="text", outputs="text").launch()
+gr.Interface(fn=summarize_dialog, 
+             inputs=["text", 
+                        gr.Slider(32, 256, value=128)
+                        ],
+             outputs="text").launch()