Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ pipeline = transformers.pipeline(
|
|
16 |
)
|
17 |
|
18 |
# Define the function to generate text
|
19 |
-
def generate_text(input_text, max_new_tokens
|
20 |
output = pipeline(
|
21 |
input_text,
|
22 |
max_new_tokens=max_new_tokens,
|
@@ -30,11 +30,11 @@ def generate_text(input_text, max_new_tokens=128, temperature=1.0, top_k=50, top
|
|
30 |
iface = gr.Interface(
|
31 |
fn=generate_text,
|
32 |
inputs=[
|
33 |
-
gr.
|
34 |
-
gr.
|
35 |
-
gr.
|
36 |
-
gr.
|
37 |
-
gr.
|
38 |
],
|
39 |
outputs="text",
|
40 |
title="Text Generation Interface",
|
|
|
16 |
)
|
17 |
|
18 |
# Define the function to generate text
|
19 |
+
def generate_text(input_text, max_new_tokens, temperature, top_k, top_p):
|
20 |
output = pipeline(
|
21 |
input_text,
|
22 |
max_new_tokens=max_new_tokens,
|
|
|
30 |
iface = gr.Interface(
|
31 |
fn=generate_text,
|
32 |
inputs=[
|
33 |
+
gr.Textbox(lines=2, placeholder="Enter your text here...", label="Input Text"),
|
34 |
+
gr.Slider(minimum=1, maximum=200, value=128, step=1, label="Max New Tokens"),
|
35 |
+
gr.Slider(minimum=0.1, maximum=2.0, value=1.0, step=0.1, label="Temperature"),
|
36 |
+
gr.Slider(minimum=1, maximum=100, value=50, step=1, label="Top-k"),
|
37 |
+
gr.Slider(minimum=0.0, maximum=1.0, value=0.95, step=0.01, label="Top-p")
|
38 |
],
|
39 |
outputs="text",
|
40 |
title="Text Generation Interface",
|