ddlei0423 commited on
Commit
cdcb299
·
verified ·
1 Parent(s): 0123537

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
  from transformers import AutoTokenizer, AutoModelForCausalLM
3
 
4
- # 模型設定
5
  model_name = "ckiplab/gpt2-base-chinese"
6
  tokenizer = AutoTokenizer.from_pretrained(model_name)
7
  model = AutoModelForCausalLM.from_pretrained(model_name)
@@ -35,4 +35,4 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue")) as demo:
35
  msg.submit(reply_fn, [state, msg], [chatbot, msg])
36
  clear.click(lambda: ([], ""), None, [chatbot, msg, state])
37
 
38
- demo.launch()
 
1
  import gradio as gr
2
  from transformers import AutoTokenizer, AutoModelForCausalLM
3
 
4
+ # 模型設定(繁體中文 GPT2)
5
  model_name = "ckiplab/gpt2-base-chinese"
6
  tokenizer = AutoTokenizer.from_pretrained(model_name)
7
  model = AutoModelForCausalLM.from_pretrained(model_name)
 
35
  msg.submit(reply_fn, [state, msg], [chatbot, msg])
36
  clear.click(lambda: ([], ""), None, [chatbot, msg, state])
37
 
38
+ demo.launch()