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__":
|