hendrydong commited on
Commit
8d099da
·
1 Parent(s): f075841

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -73,19 +73,19 @@ css = """
73
  @dataclass
74
  class ChatbotArguments:
75
  prompt_structure: Optional[str] = field(
76
- default="{input_text}",
77
  metadata={
78
  "help": "prompt structure given user's input text"
79
  },
80
  )
81
  end_string: Optional[str] = field(
82
- default="\n\n",
83
  metadata={
84
  "help": "end string mark of the chatbot's output"
85
  },
86
  )
87
  max_new_tokens: Optional[int] = field(
88
- default=1800,
89
  metadata={
90
  "help": "maximum number of generated tokens"
91
  },
 
73
  @dataclass
74
  class ChatbotArguments:
75
  prompt_structure: Optional[str] = field(
76
+ default="###Human: {input_text}###Assistant:",
77
  metadata={
78
  "help": "prompt structure given user's input text"
79
  },
80
  )
81
  end_string: Optional[str] = field(
82
+ default="#",
83
  metadata={
84
  "help": "end string mark of the chatbot's output"
85
  },
86
  )
87
  max_new_tokens: Optional[int] = field(
88
+ default=1500,
89
  metadata={
90
  "help": "maximum number of generated tokens"
91
  },