Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -24,8 +24,8 @@ def generate_text(prompt, stop_choice):
|
|
24 |
iface = gr.Interface(
|
25 |
fn=generate_text,
|
26 |
inputs=[
|
27 |
-
gr.
|
28 |
-
gr.
|
29 |
],
|
30 |
outputs="text",
|
31 |
title="DistilGPT2 Text Generation",
|
|
|
24 |
iface = gr.Interface(
|
25 |
fn=generate_text,
|
26 |
inputs=[
|
27 |
+
gr.Textbox(lines=4, placeholder="Enter your prompt here...", label="Prompt"),
|
28 |
+
gr.Dropdown(choices=["Period", "Space"], label="Stop Token")
|
29 |
],
|
30 |
outputs="text",
|
31 |
title="DistilGPT2 Text Generation",
|