teragron commited on
Commit
9752ddb
1 Parent(s): a53bf97

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -7
app.py CHANGED
@@ -20,14 +20,17 @@ def chatbot(prompt, temperature, topt, maxtoken, model_file):
20
 
21
  with gr.Blocks() as demo:
22
  gr.Markdown("HF Spaces for TinyStories")
23
- with gr.Row():
24
- inp = gr.Textbox(placeholder="Type the beginning of the story")
25
  with gr.Column():
26
- temperature_slider = gr.Slider(minimum=0.1, maximum=2.0, value=0.8, label="Temperature")
27
- topt_slider = gr.Slider(minimum=0.1, maximum=1.0, value=0.9, label="Topt")
28
- maxtoken_slider = gr.Slider(minimum=16, maximum=512, value=256, label="Max Tokens")
29
- model_file_dropdown = gr.Dropdown(choices=["stories15M.bin", "stories42M.bin", "stories110M.bin"], label="Model File")
30
- out = gr.Textbox()
 
 
 
31
  btn = gr.Button("Run")
32
  btn.click(fn=chatbot, inputs=[inp, temperature_slider, topt_slider, maxtoken_slider, model_file_dropdown], outputs=out)
33
 
 
20
 
21
  with gr.Blocks() as demo:
22
  gr.Markdown("HF Spaces for TinyStories")
23
+
24
+
25
  with gr.Column():
26
+ inp = gr.Textbox(placeholder="Type the beginning of the story")
27
+ with gr.Row():
28
+ with gr.Column():
29
+ temperature_slider = gr.Slider(minimum=0.1, maximum=2.0, value=0.8, label="Temperature")
30
+ topt_slider = gr.Slider(minimum=0.1, maximum=1.0, value=0.9, label="Topt")
31
+ maxtoken_slider = gr.Slider(minimum=16, maximum=512, value=256, label="Max Tokens")
32
+ model_file_dropdown = gr.Dropdown(choices=["stories15M.bin", "stories42M.bin", "stories110M.bin"], label="Model File", value="stories15M.bin")
33
+ out = gr.Textbox()
34
  btn = gr.Button("Run")
35
  btn.click(fn=chatbot, inputs=[inp, temperature_slider, topt_slider, maxtoken_slider, model_file_dropdown], outputs=out)
36