Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -11,7 +11,7 @@ def generate(prompt):
|
|
11 |
result_length = len(prompt) + 4
|
12 |
inputs = tokenizer(prompt, return_tensors="pt") # 回傳的張量使用 Pytorch的格式。如果是 Tensorflow 格式的話,則指定為 "tf"。
|
13 |
results = model.generate(inputs["input_ids"],
|
14 |
-
num_return_sequences=
|
15 |
max_length=result_length,
|
16 |
early_stopping=True,
|
17 |
do_sample=True,
|
@@ -33,11 +33,11 @@ with gr.Blocks() as demo:
|
|
33 |
""")
|
34 |
with gr.Row():
|
35 |
with gr.Column():
|
36 |
-
result = gr.components.Textbox(lines=7, label="
|
37 |
-
prompt = gr.components.Textbox(lines=2, label="
|
38 |
-
gr.Examples(examples, label='
|
39 |
prompt.change(generate, inputs=[prompt], outputs=[result])
|
40 |
-
btn = gr.Button("
|
41 |
btn.click(generate, inputs=[result], outputs=[result])
|
42 |
|
43 |
if __name__ == "__main__":
|
|
|
11 |
result_length = len(prompt) + 4
|
12 |
inputs = tokenizer(prompt, return_tensors="pt") # 回傳的張量使用 Pytorch的格式。如果是 Tensorflow 格式的話,則指定為 "tf"。
|
13 |
results = model.generate(inputs["input_ids"],
|
14 |
+
num_return_sequences=2, # 產生 2 個句子回來。
|
15 |
max_length=result_length,
|
16 |
early_stopping=True,
|
17 |
do_sample=True,
|
|
|
33 |
""")
|
34 |
with gr.Row():
|
35 |
with gr.Column():
|
36 |
+
result = gr.components.Textbox(lines=7, label="Generative")
|
37 |
+
prompt = gr.components.Textbox(lines=2, label="Prompt", placeholder=examples[0], visible=False)
|
38 |
+
gr.Examples(examples, label='Examples', inputs=[prompt])
|
39 |
prompt.change(generate, inputs=[prompt], outputs=[result])
|
40 |
+
btn = gr.Button("Next sentence")
|
41 |
btn.click(generate, inputs=[result], outputs=[result])
|
42 |
|
43 |
if __name__ == "__main__":
|