KantaHayashiAI commited on
Commit
4b29b18
·
verified ·
1 Parent(s): 792aadd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +21 -6
app.py CHANGED
@@ -68,12 +68,27 @@ def generate(
68
  demo = gr.ChatInterface(
69
  fn=generate,
70
  additional_inputs=[
71
- gr.Slider("Max new tokens", minimum=1, maximum=MAX_MAX_NEW_TOKENS,
72
- step=1, value=DEFAULT_MAX_NEW_TOKENS),
73
- gr.Slider("Temperature", minimum=0.1, maximum=4.0,
74
- step=0.1, value=0.6),
75
- gr.Slider("Top-p (nucleus sampling)", minimum=0.05,
76
- maximum=1.0, step=0.05, value=0.9),
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
77
  ],
78
  stop_btn=None,
79
  examples=[["Write me an English pangram."]],
 
68
  demo = gr.ChatInterface(
69
  fn=generate,
70
  additional_inputs=[
71
+ gr.Slider(
72
+ label="Max new tokens",
73
+ minimum=1,
74
+ maximum=MAX_MAX_NEW_TOKENS,
75
+ step=1,
76
+ value=DEFAULT_MAX_NEW_TOKENS,
77
+ ),
78
+ gr.Slider(
79
+ label="Temperature",
80
+ minimum=0.1,
81
+ maximum=4.0,
82
+ step=0.1,
83
+ value=0.6,
84
+ ),
85
+ gr.Slider(
86
+ label="Top-p (nucleus sampling)",
87
+ minimum=0.05,
88
+ maximum=1.0,
89
+ step=0.05,
90
+ value=0.9,
91
+ ),
92
  ],
93
  stop_btn=None,
94
  examples=[["Write me an English pangram."]],