Update app.py

app.py

@@ -22,7 +22,7 @@ def get_model_pipeline(model_name):
 
 # Function to generate code
 @st.cache_data
-def generate_code(task_description, max_length, temperature, num_return_sequences, model_name):
+def generate_code(task_description, temperature, num_return_sequences, model_name):
     code_pipeline = get_model_pipeline(model_name)
     if code_pipeline is None:
         return ["Error: Failed to load model pipeline."]
@@ -31,7 +31,7 @@ def generate_code(task_description, max_length, temperature, num_return_sequence
         logging.info(f"Generating code with input: {task_description}")
         prompt = f"Develop code for the following task: {task_description}"
         
-        outputs = code_pipeline(prompt, max_length=max_length, num_return_sequences=num_return_sequences, temperature=temperature)
+        outputs = code_pipeline(prompt, temperature=temperature, num_return_sequences=num_return_sequences)
         codes = [output['generated_text'] for output in outputs]
         
         logging.info("Code generation completed successfully.")