Spaces:
Running
Running
Update app.py
Browse files
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 |
-
|
24 |
-
|
25 |
with gr.Column():
|
26 |
-
|
27 |
-
|
28 |
-
|
29 |
-
|
30 |
-
|
|
|
|
|
|
|
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 |
|