YanshekWoo commited on
Commit
7a75ce5
1 Parent(s): 9e02690

add requirements.txt

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. requirements.txt +3 -0
app.py CHANGED
@@ -4,8 +4,8 @@ from transformers import BertTokenizer, BartForConditionalGeneration
4
 
5
  title = "HIT-TMG/dialogue-bart-large-chinese"
6
  description = """
7
- This is a seq2seq model fine-tuned on several Chinese dialogue datasets, from bart-large-chinese. \n
8
- See some details of model card at https://huggingface.co/HIT-TMG/dialogue-bart-large-chinese . \n\n
9
  Besides starting the conversation from scratch, you can also input the whole dialogue history utterance by utterance seperated by '[SEP]'. \n
10
  (e.g. "可以认识一下吗[SEP]当然可以啦,你好。[SEP]嘿嘿你好,请问你最近在忙什么呢?[SEP]我最近养了一只狗狗,我在训练它呢。") \n
11
  """
 
4
 
5
  title = "HIT-TMG/dialogue-bart-large-chinese"
6
  description = """
7
+ This is a fine-tuned version of HIT-TMG/dialogue-bart-large-chinese on the DuSinc dataset. \n
8
+ See some details of model card at https://huggingface.co/HIT-TMG/dialogue-bart-large-chinese-DuSinc . \n\n
9
  Besides starting the conversation from scratch, you can also input the whole dialogue history utterance by utterance seperated by '[SEP]'. \n
10
  (e.g. "可以认识一下吗[SEP]当然可以啦,你好。[SEP]嘿嘿你好,请问你最近在忙什么呢?[SEP]我最近养了一只狗狗,我在训练它呢。") \n
11
  """
requirements.txt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ gradio
2
+ torch
3
+ transformers