Update app.py

app.py

@@ -20,7 +20,7 @@ def infer(input_ids, max_length, temperature, top_k, top_p, num_return_sequences
         top_k=top_k,
         top_p=top_p,
         do_sample=True,
-        num_return_sequences=num_return_sequences,
+        num_return_sequences=1
     )
 
     return output_sequences
@@ -35,8 +35,6 @@ max_length = st.sidebar.slider("Max Length", min_value = 10, max_value=30)
 temperature = st.sidebar.slider("Temperature", value = 1.0, min_value = 0.0, max_value=1.0, step=0.05)
 top_k = st.sidebar.slider("Top-k", min_value = 0, max_value=5, value = 0)
 top_p = st.sidebar.slider("Top-p", min_value = 0.0, max_value=1.0, step = 0.05, value = 0.9)
-num_return_sequences = st.sidebar.number_input('Number of Return Sequences', min_value=1, max_value=5, value=1, step=1)
-
 
 encoded_prompt = tokenizer.encode(sent, add_special_tokens=False, return_tensors="pt")
 if encoded_prompt.size()[-1] == 0: