teragron commited on
Commit
e5ce199
1 Parent(s): f7a5415

Update app.py

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