suriya7 commited on
Commit
73f027a
·
verified ·
1 Parent(s): 39caeca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -9
app.py CHANGED
@@ -81,14 +81,7 @@ with gr.Blocks() as demo:
81
  interactive=True,
82
  label="Max New Tokens",
83
  )
84
- top_p = gr.Slider(
85
- minimum=0.05,
86
- maximum=1.0,
87
- value=0.95,
88
- step=0.05,
89
- interactive=True,
90
- label="Top-p (nucleus sampling)",
91
- )
92
  top_k = gr.Slider(
93
  minimum=1,
94
  maximum=50,
@@ -108,7 +101,7 @@ with gr.Blocks() as demo:
108
 
109
  user_text.submit(
110
  run_generation,
111
- [user_text, top_p, temperature, top_k, max_new_tokens],
112
  model_output,
113
  )
114
  button_submit.click(
 
81
  interactive=True,
82
  label="Max New Tokens",
83
  )
84
+
 
 
 
 
 
 
 
85
  top_k = gr.Slider(
86
  minimum=1,
87
  maximum=50,
 
101
 
102
  user_text.submit(
103
  run_generation,
104
+ [user_text,repetition_penalty=1.5,temperature,do_sample=True,top_k, max_new_tokens],
105
  model_output,
106
  )
107
  button_submit.click(