jslin09 commited on
Commit
315025d
1 Parent(s): dccf4fa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
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=5, # 產生五個句子回來。
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="生成的草稿", show_label=True)
37
- prompt = gr.components.Textbox(lines=2, label="輸入提示文字", placeholder=examples[0],visible=False)
38
- gr.Examples(examples, label='例句', inputs=[prompt])
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__":