Update app.py

app.py

@@ -44,8 +44,8 @@ def response(message = "Hello, I'm a language model", num_return_sequences = 5,m
 
 
 # Function to generate text
-def generate_text(prompt, top_k, max_return_sequences, max_tokens):
-    ans = response(message = prompt, num_return_sequences = max_return_sequences,max_length = max_tokens,top_k = top_k)
+def generate_text(prompt, top_k, return_sequences, max_tokens):
+    ans = response(message = prompt, num_return_sequences = return_sequences,max_length = max_tokens,top_k = top_k)
     
 
     return ans