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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -11
app.py CHANGED
@@ -21,17 +21,16 @@ def chatbot(prompt, temperature, topt, maxtoken, model_file):
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
 
37
  demo.launch()
 
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
+ btn = gr.Button("Run")
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.click(fn=chatbot, inputs=[inp, temperature_slider, topt_slider, maxtoken_slider, model_file_dropdown], outputs=out)
35
 
36
  demo.launch()