Neu256 commited on
Commit
cec4192
·
verified ·
1 Parent(s): d4379ad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,5 +41,5 @@ def model_generate(text, number_of_new_token, temperature, top_p):
41
 
42
  return decode(enc_sec=m.generate(idx=context, max_new_tokens=number_of_new_token, temperature = temperature, top_p = top_p)[0], tokenizer=tokenizer)
43
 
44
- 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")
45
  iface.launch()
 
41
 
42
  return decode(enc_sec=m.generate(idx=context, max_new_tokens=number_of_new_token, temperature = temperature, top_p = top_p)[0], tokenizer=tokenizer)
43
 
44
+ iface = gr.Interface(fn=model_generate, inputs=["text", gr.Slider(10, 1000), gr.Slider(0, 1, value=0.7, step = 0.05), gr.Slider(0, 1, value=0.95, step = 0.05)], outputs="text")
45
  iface.launch()