Spaces:
Runtime error
Runtime error
Commit
·
fb0f59e
1
Parent(s):
be757b7
Update app.py
Browse files
app.py
CHANGED
@@ -12,7 +12,7 @@ generator = pipeline(
|
|
12 |
)
|
13 |
model = AutoModelWithLMHead.from_pretrained("gpt2")
|
14 |
def green(prompt):
|
15 |
-
return generator(prompt, max_length=
|
16 |
|
17 |
iface = gr.Interface(fn=green, inputs=gr.inputs.Textbox(lines=2, placeholder="Put prompt here"), outputs=gr.outputs.Textbox(type="str", label="generated text"))
|
18 |
iface.launch()
|
|
|
12 |
)
|
13 |
model = AutoModelWithLMHead.from_pretrained("gpt2")
|
14 |
def green(prompt):
|
15 |
+
return generator(prompt, max_length=1000)
|
16 |
|
17 |
iface = gr.Interface(fn=green, inputs=gr.inputs.Textbox(lines=2, placeholder="Put prompt here"), outputs=gr.outputs.Textbox(type="str", label="generated text"))
|
18 |
iface.launch()
|