Update app.py

app.py

@@ -36,10 +36,10 @@ m.eval()
 #print(
 #    "Model with {:.2f}M parameters".format(sum(p.numel() for p in m.parameters()) / 1e6)
 #)
-def model_generate(text, number, TEMPERATURE, TOP_K):
+def model_generate(text, number_of_new_token, temperature, top_p):
     context = encode(str(text), tokenizer).unsqueeze(0).to(DEVICE)
 
-    return decode(enc_sec=m.generate(idx=context, max_new_tokens=number, temperature = TEMPERATURE, top_k = TOP_K)[0], tokenizer=tokenizer)
+    return decode(enc_sec=m.generate(idx=context, max_new_tokens=number_of_new_token, temperature = temperature, top_p = top_p)[0], tokenizer=tokenizer)
 
-iface = gr.Interface(fn=model_generate, inputs=["text", gr.Slider(10, 1000), gr.Slider(0, 1), gr.Slider(0, 1)], outputs="text")
+iface = gr.Interface(fn=model_generate, inputs=["text", gr.Slider(10, 1000), gr.Slider(0, 1, value=0.7, step = 0.5), gr.Slider(0, 1, value=0.95, step = 0.5)], outputs="text")
 iface.launch()