Add sequences input

app.py

@@ -35,8 +35,8 @@ fit_cbs = [Seq2SeqMetricsCallback(custom_metrics=summarization_metrics, calc_eve
 
 learn = load_learner(fname='cnn_model.pkl')
 
-def get_summary(text):
-    return learn.blurr_summarize(text, num_return_sequences=5)
+def get_summary(text, sequences_num):
+    return learn.blurr_summarize(text, num_return_sequences=sequences_num)
 
-iface = gr.Interface(fn=get_summary, inputs="text", outputs="text")
+iface = gr.Interface(fn=get_summary, inputs=["text", gr.Number(value=5, label="sequences")], outputs="text")
 iface.launch()