Update app.py

app.py

@@ -19,6 +19,7 @@ def process(text: str,
             max_len: int,
             temp: float,
             top_k: int,
+            num_return_sequences:int,
             top_p: float):
     payload = {
         "inputs": text,
@@ -27,6 +28,7 @@ def process(text: str,
             "top_k": top_k,
             "top_p": top_p,
             "temperature": temp,
+            "num_return_sequences": 10,
             "repetition_penalty": 2.0,
         },
         "options": {
@@ -54,6 +56,11 @@ top_k = st.sidebar.text_input(
     value=50,
     help="The number of highest probability vocabulary tokens to keep for top-k-filtering."
 )
+num_return_sequences = st.sidebar.text_input(
+    "Returns",
+    value=5,
+    help="Number to return."
+)
 top_p = st.sidebar.text_input(
     "Top p",
     value=0.95,