pszemraj commited on
Commit
2318c3b
1 Parent(s): 334ede4

adjust generation params

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -46,7 +46,7 @@ my_cwd = str(cwd.resolve()) # string so it can be passed to os.path() objects
46
 
47
 
48
  def chat(
49
- prompt_message, temperature: float = 0.6, top_p: float = 0.95, top_k: int = 25
50
  ) -> str:
51
  """
52
  chat - the main function for the chatbot. This is the function that is called when the user
@@ -81,10 +81,10 @@ def ask_gpt(
81
  speaker="person alpha",
82
  responder="person beta",
83
  min_length=4,
84
- max_length=64,
85
  top_p=0.95,
86
  top_k=25,
87
- temperature=0.6,
88
  ) -> str:
89
  """
90
  ask_gpt - helper function that asks the GPT model a question and returns the response
@@ -223,15 +223,15 @@ if __name__ == "__main__":
223
  examples_per_page=8,
224
  examples=[
225
  ["Point Break or Bad Boys II?", 0.75, 0.95, 50],
226
- ["So... you're saying this wasn't an accident?", 0.6, 0.95, 50],
227
  ["Hi, my name is Reginald", 0.6, 0.95, 100],
228
  ["Happy birthday!", 0.9, 0.95, 50],
229
  ["I have a question, can you help me?", 0.6, 0.95, 50],
230
  ["Do you know a joke?", 0.8, 0.85, 50],
231
  ["Will you marry me?", 0.9, 0.95, 100],
232
- ["Are you single?", 0.6, 0.95, 100],
233
  ["Do you like people?", 0.7, 0.95, 25],
234
- ["You never took a short cut before?", 0.7, 0.95, 100],
235
  ],
236
  title=f"GPT Chatbot Demo: {default_model} Model",
237
  description=f"A Demo of a Chatbot trained for conversation with humans. Size XL= 1.5B parameters.\n\n"
 
46
 
47
 
48
  def chat(
49
+ prompt_message, temperature: float = 0.5, top_p: float = 0.95, top_k: int = 20
50
  ) -> str:
51
  """
52
  chat - the main function for the chatbot. This is the function that is called when the user
 
81
  speaker="person alpha",
82
  responder="person beta",
83
  min_length=4,
84
+ max_length=48,
85
  top_p=0.95,
86
  top_k=25,
87
+ temperature=0.5,
88
  ) -> str:
89
  """
90
  ask_gpt - helper function that asks the GPT model a question and returns the response
 
223
  examples_per_page=8,
224
  examples=[
225
  ["Point Break or Bad Boys II?", 0.75, 0.95, 50],
226
+ ["So... you're saying this wasn't an accident?", 0.6, 0.95, 40],
227
  ["Hi, my name is Reginald", 0.6, 0.95, 100],
228
  ["Happy birthday!", 0.9, 0.95, 50],
229
  ["I have a question, can you help me?", 0.6, 0.95, 50],
230
  ["Do you know a joke?", 0.8, 0.85, 50],
231
  ["Will you marry me?", 0.9, 0.95, 100],
232
+ ["Are you single?", 0.95, 0.95, 100],
233
  ["Do you like people?", 0.7, 0.95, 25],
234
+ ["You never took a shortcut before?", 0.7, 0.95, 100],
235
  ],
236
  title=f"GPT Chatbot Demo: {default_model} Model",
237
  description=f"A Demo of a Chatbot trained for conversation with humans. Size XL= 1.5B parameters.\n\n"