saiteki-kai commited on
Commit
0727ece
·
verified ·
1 Parent(s): f9bf7bf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -17
app.py CHANGED
@@ -5,10 +5,10 @@ from huggingface_hub import InferenceClient
5
  def respond(
6
  message,
7
  history: list[dict[str, str]],
8
- system_message,
9
- max_tokens,
10
- temperature,
11
- top_p,
12
  hf_token: gr.OAuthToken,
13
  ):
14
  """
@@ -18,7 +18,7 @@ def respond(
18
 
19
  messages = [{"role": "system", "content": system_message}]
20
 
21
- messages.extend(history)
22
 
23
  messages.append({"role": "user", "content": message})
24
 
@@ -46,18 +46,18 @@ For information on how to customize the ChatInterface, peruse the gradio docs: h
46
  chatbot = gr.ChatInterface(
47
  respond,
48
  type="messages",
49
- additional_inputs=[
50
- gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
51
- gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
52
- gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
53
- gr.Slider(
54
- minimum=0.1,
55
- maximum=1.0,
56
- value=0.95,
57
- step=0.05,
58
- label="Top-p (nucleus sampling)",
59
- ),
60
- ],
61
  )
62
 
63
  with gr.Blocks() as demo:
 
5
  def respond(
6
  message,
7
  history: list[dict[str, str]],
8
+ system_message="Your are an helpful assistant",
9
+ max_tokens=2048,
10
+ temperature=0.0,
11
+ top_p=None,
12
  hf_token: gr.OAuthToken,
13
  ):
14
  """
 
18
 
19
  messages = [{"role": "system", "content": system_message}]
20
 
21
+ # messages.extend(history)
22
 
23
  messages.append({"role": "user", "content": message})
24
 
 
46
  chatbot = gr.ChatInterface(
47
  respond,
48
  type="messages",
49
+ #additional_inputs=[
50
+ # gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
51
+ # gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
52
+ # gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
53
+ # gr.Slider(
54
+ # minimum=0.1,
55
+ # maximum=1.0,
56
+ # value=0.95,
57
+ # step=0.05,
58
+ # label="Top-p (nucleus sampling)",
59
+ # ),
60
+ #],
61
  )
62
 
63
  with gr.Blocks() as demo: