Update app.py
Browse files
app.py
CHANGED
@@ -33,9 +33,9 @@ def generate(text = "", max_new_tokens = 128):
|
|
33 |
demo = gr.Interface(
|
34 |
title="TextIteratorStreamer + Gradio demo",
|
35 |
fn=generate,
|
36 |
-
inputs=[gr.
|
37 |
-
gr.
|
38 |
-
outputs=gr.
|
39 |
allow_flagging="never"
|
40 |
)
|
41 |
|
|
|
33 |
demo = gr.Interface(
|
34 |
title="TextIteratorStreamer + Gradio demo",
|
35 |
fn=generate,
|
36 |
+
inputs=[gr.Textbox(lines=5, label="Input Text"),
|
37 |
+
gr.Slider(default=128,minimum=5, maximum=256, step=1, label="Maximum number of new tokens")],
|
38 |
+
outputs=gr.Textbox(label="Generated Text"),
|
39 |
allow_flagging="never"
|
40 |
)
|
41 |
|