Tonic commited on
Commit
978d21a
·
verified ·
1 Parent(s): e489c90

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -100,11 +100,11 @@ with gr.Blocks(theme=gr.themes.Base()) as iface:
100
 
101
  prompt = gr.Textbox(label="Add a passage in the style of historical texts", placeholder="Hi there my name is Tonic and I ride my bicycle along the river Seine:", lines=3)
102
 
103
- max_new_tokens = gr.Slider(label="Max New Tokens", minimum=50, maximum=1000, step=10, value=140)
104
- top_k = gr.Slider(label="Top-k Sampling", minimum=1, maximum=100, step=0.05, value=50)
105
- temperature = gr.Slider(label="Temperature", minimum=0.1, maximum=1.5, step=0.05, value=0.3)
106
- top_p = gr.Slider(label="Top-p (Nucleus Sampling)", minimum=0.1, maximum=1.0, step=0.005, value=0.95)
107
- repetition_penalty = gr.Slider(label="Repetition Penalty", minimum=0.5, maximum=2.0, step=0.05, value=1.0)
108
 
109
  generated_text_output = gr.Textbox(label="🎅🏻⌚OCRonos-Vintage")
110
  highlighted_text = gr.HighlightedText(label="🎅🏻⌚Tokenized", combine_adjacent=True, show_legend=True)
 
100
 
101
  prompt = gr.Textbox(label="Add a passage in the style of historical texts", placeholder="Hi there my name is Tonic and I ride my bicycle along the river Seine:", lines=3)
102
 
103
+ max_new_tokens = gr.Slider(label="Max New Tokens", minimum=50, maximum=1000, step=5, value=90)
104
+ top_k = gr.Slider(label="Top-k Sampling", minimum=1, maximum=100, step=1, value=50)
105
+ temperature = gr.Slider(label="Temperature", minimum=0.1, maximum=1, step=0.05, value=0.5)
106
+ top_p = gr.Slider(label="Top-p (Nucleus Sampling)", minimum=0.1, maximum=0.99, step=0.01, value=0.97)
107
+ repetition_penalty = gr.Slider(label="Repetition Penalty", minimum=0.5, maximum=2.0, step=0.05, value=1.3)
108
 
109
  generated_text_output = gr.Textbox(label="🎅🏻⌚OCRonos-Vintage")
110
  highlighted_text = gr.HighlightedText(label="🎅🏻⌚Tokenized", combine_adjacent=True, show_legend=True)