Update app.py

app.py

@@ -12,7 +12,7 @@ def generate_code(prompt, max_length):
     inputs = tokenizer(prompt, return_tensors="pt")
     streamer = TextIteratorStreamer(tokenizer)
     
-    generation_kwargs = dict(inputs, streamer=streamer, max_length=max_length)
+    generation_kwargs = dict(inputs, streamer=streamer, max_new_tokens=int(max_length))
     thread = Thread(target=model.generate, kwargs=generation_kwargs)
     thread.start()
 
@@ -32,8 +32,8 @@ iface = gr.Interface(
     title="Qwen2.5-Coder-1.5B Code Generator",
     description="Enter a prompt to generate Python code using Qwen2.5-Coder-1.5B",
     examples=[
-        ["Write a Python function to calculate the factorial of a number."],
-        ["Create a class representing a simple bank account with deposit and withdraw methods."]
+        ["Write a Python function to calculate the factorial of a number.", 200],
+        ["Create a class representing a simple bank account with deposit and withdraw methods.", 300]
     ]
 )