fix output format

app.py

@@ -5,13 +5,16 @@ from transformers import pipeline
 summarizer = pipeline('summarization', model='Eitanli/resume_label_summary_model')
 
 
-def predict(text):
-    summary = summarizer(text, max_length=10, min_length=2)
+def predict(text, min_length, max_length):
+    summary = summarizer(text, min_length=min_length, max_length=max_length)
     return summary[0]['summary_text']
 
 
 iface = gr.Interface(
     fn=predict,
-    inputs='text',
+    inputs=[
+        gr.Textbox(label="Paste resume here"),
+        gr.Slider(2, 20, step=1.0, value=2, label="min_length", info="Choose between 2 and 20"),
+        gr.Slider(10, 30, step=1.0, value=10, label="max_length", info="Choose between 10 and 30")],
     outputs="text")
 iface.launch()