Update app.py

app.py

@@ -15,7 +15,7 @@ def load_model(model_name):
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     model = AutoModelForCausalLM.from_pretrained(model_name)
     return model, tokenizer
-def extend(input_text, max_size=20, top_k=50, top_p=0.95, num_return_sequences):
+def extend(input_text, num_return_sequences, max_size=20, top_k=50, top_p=0.95):
     if len(input_text) == 0:
         input_text = ""
     encoded_prompt = tokenizer.encode(