Spaces:
Running
Running
Update app.py
Browse files
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 |
-
|
| 25 |
-
|
| 26 |
-
|
| 27 |
-
|
| 28 |
-
|
| 29 |
-
|
| 30 |
-
|
| 31 |
-
|
| 32 |
-
|
| 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 |
|