Update app.py

app.py

@@ -12,13 +12,11 @@ model = AutoModelForCausalLM.from_pretrained("NovelAI/genji-python-6B").half().e
 import gradio as gr
 
 maxLength=400
-temperature=0.3
 top_k = 50
-top_p = 0.9
 repetition_penalty = 1.13
 repetition_penalty_range = 512
 repetition_penalty_slope = 3.33
-def generator(text):
+def generator(text, temperature,top_p):
     tokens = tokenizer(text, return_tensors="pt").input_ids.cuda()[:, -(2047-maxLength):]
     out = model.generate(
         tokens.long(),
@@ -43,11 +41,14 @@ article = "<p style='text-align: center'><a href='https://colab.research.google.
 
 gr.Interface(
     generator, 
-    gr.inputs.Textbox(label="input text", lines=5), 
+    [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.outputs.Textbox(label="Output text"),
     title=title,
     description=description,
     article=article,
     examples=[
-        ['def print_customer_name']
+        ['def print_Hello_Huggingface():']
     ]).launch(debug=True)