Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -12,7 +12,7 @@ pipeline = transformers.pipeline(
|
|
12 |
)
|
13 |
|
14 |
# Define the function to generate text
|
15 |
-
def generate_text(input_text, max_new_tokens, temperature,
|
16 |
output = pipeline(
|
17 |
input_text,
|
18 |
max_new_tokens=max_new_tokens,
|
@@ -72,4 +72,4 @@ with gr.Blocks() as iface:
|
|
72 |
|
73 |
# Launch the interface
|
74 |
if __name__ == "__main__":
|
75 |
-
iface.launch()
|
|
|
12 |
)
|
13 |
|
14 |
# Define the function to generate text
|
15 |
+
def generate_text(input_text, max_new_tokens, temperature, top_p, split_output):
|
16 |
output = pipeline(
|
17 |
input_text,
|
18 |
max_new_tokens=max_new_tokens,
|
|
|
72 |
|
73 |
# Launch the interface
|
74 |
if __name__ == "__main__":
|
75 |
+
iface.launch()
|