Update app.py

app.py

@@ -8,9 +8,9 @@ from transformers import pipeline
 code_writer = pipeline("text-generation", model="Qwen/Qwen2.5-Coder-7B-Instruct",torch_dtype=torch.bfloat16 )
 
 # Define the code generation function
-def generate_code(prompt, max_length):
-    # Generate code based on the input prompt and maximum length
-    output = code_writer(prompt, max_length=max_length, num_return_sequences=1, do_sample=True)
+def generate_code(prompt):
+    # Generate code based on the input prompt
+    output = code_writer(prompt, num_return_sequences=1, do_sample=True)
     return output[0]['generated_text']
 
 # Close any existing Gradio instances
@@ -20,11 +20,7 @@ gr.close_all()
 Demo = gr.Interface(
     fn=generate_code,
     inputs=[
-        gr.Textbox(label="Code Prompt", lines=5, placeholder="Enter a description or snippet for code generation."),
-        gr.Slider(
-            label="Code Length (Tokens Approx.)", 
-            minimum=10, maximum=2000, step=20, value=50
-        )
+        gr.Textbox(label="Code Prompt", lines=5, placeholder="Enter a description or snippet for code generation.")
     ],
     outputs=[gr.Textbox(label="Generated Code", lines=10)],
     title="Code_Generator_App",
@@ -33,4 +29,3 @@ Demo = gr.Interface(
 
 # Launch the app with a public link
 Demo.launch(share=True)
-