Update README.md: make it directly usable from huggingface preview ui

#1
Files changed (1) hide show
  1. README.md +4 -3
README.md CHANGED
@@ -1,8 +1,9 @@
1
  ---
2
  language: zh
3
- widget:
4
- - text: "你们宿舍都是这么厉害的人吗"
5
  license: apache-2.0
 
6
  ---
7
 
8
  # liam168/chat-DialoGPT-small-zh
@@ -36,4 +37,4 @@ for step in range(5):
36
 
37
  # pretty print last ouput tokens from bot
38
  print("Answer: {}".format(tokenizer.decode(chat_history_ids[:, bot_input_ids.shape[-1]:][0], skip_special_tokens=True)))
39
- ```
 
1
  ---
2
  language: zh
3
+ widget:
4
+ - text: 你们宿舍都是这么厉害的人吗
5
  license: apache-2.0
6
+ pipeline_tag: conversational
7
  ---
8
 
9
  # liam168/chat-DialoGPT-small-zh
 
37
 
38
  # pretty print last ouput tokens from bot
39
  print("Answer: {}".format(tokenizer.decode(chat_history_ids[:, bot_input_ids.shape[-1]:][0], skip_special_tokens=True)))
40
+ ```