Update app.py

app.py

@@ -10,7 +10,7 @@ model = AutoModelWithLMHead.from_pretrained("gpt2-large")
 
 
 
-def infer(input_ids, max_length, temperature, top_k, top_p, do_sample = True, num_return_sequences):
+def infer(input_ids, max_length, temperature, top_k, top_p, num_return_sequences):
 
     output_sequences = model.generate(
         input_ids=input_ids,
@@ -44,7 +44,7 @@ else:
     input_ids = encoded_prompt
 
 
-output_sequences = infer(input_ids, max_length, temperature, top_k, top_p, do_sample = True, num_return_sequences)
+output_sequences = infer(input_ids, max_length, temperature, top_k, top_p, num_return_sequences)
 
 
 for generated_sequence_idx, generated_sequence in enumerate(output_sequences):