Update app.py
Browse files
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()
|