abubasith86 commited on
Commit
9e78176
·
1 Parent(s): 892e0a4
Files changed (1) hide show
  1. app.py +4 -17
app.py CHANGED
@@ -61,9 +61,9 @@ Your goals:
61
  def respond(
62
  message,
63
  history: list[tuple[str, str]],
64
- max_tokens,
65
- temperature,
66
- top_p,
67
  ):
68
 
69
  recent_keywords = [
@@ -147,20 +147,7 @@ def respond(
147
  """
148
  For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
149
  """
150
- demo = gr.ChatInterface(
151
- respond,
152
- additional_inputs=[
153
- gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
154
- gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
155
- gr.Slider(
156
- minimum=0.1,
157
- maximum=1.0,
158
- value=0.95,
159
- step=0.05,
160
- label="Top-p (nucleus sampling)",
161
- ),
162
- ],
163
- )
164
 
165
  if __name__ == "__main__":
166
  demo.launch()
 
61
  def respond(
62
  message,
63
  history: list[tuple[str, str]],
64
+ max_tokens=2048,
65
+ temperature=0.4,
66
+ top_p=0.1,
67
  ):
68
 
69
  recent_keywords = [
 
147
  """
148
  For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
149
  """
150
+ demo = gr.ChatInterface(respond)
 
 
 
 
 
 
 
 
 
 
 
 
 
151
 
152
  if __name__ == "__main__":
153
  demo.launch()