Update app.py

app.py

@@ -11,12 +11,11 @@ model = AutoModelForCausalLM.from_pretrained("NovelAI/genji-python-6B").half().e
 
 import gradio as gr
 
-maxLength=400
 top_k = 50
 repetition_penalty = 1.13
 repetition_penalty_range = 512
 repetition_penalty_slope = 3.33
-def generator(text, temperature,top_p):
+def generator(text, temperature ,top_p, maxLength):
     tokens = tokenizer(text, return_tensors="pt").input_ids.cuda()[:, -(2047-maxLength):]
     out = model.generate(
         tokens.long(),
@@ -44,11 +43,12 @@ gr.Interface(
     [gr.inputs.Textbox(label="input text", lines=5),
     gr.inputs.Slider(minimum=0, maximum=1, step=0.1, default=0.2, label="Temperature"),
     gr.inputs.Slider(minimum=0, maximum=1, step=0.1, default=1, label="Top P")
+    gr.inputs.Slider(minimum=1, maximum=400, step=10, default=300, label="Max Length")
     ], 
     gr.outputs.Textbox(label="Output text"),
     title=title,
     description=description,
     article=article,
     examples=[
-        ['def print_Hello_Huggingface():']
+        ['def print_Hello_Huggingface():', 0.2, 1,300]
     ]).launch(debug=True)