richardkimsm89 commited on
Commit
c7bde51
·
verified ·
1 Parent(s): 12fb4a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -3
app.py CHANGED
@@ -9,14 +9,20 @@ client = InferenceClient(model)
9
  def fn_text(
10
  prompt,
11
  history,
12
- #system_prompt,
13
  max_tokens,
14
  temperature,
15
  top_p,
16
  ):
17
- messages = [{"role": "user", "content": prompt}]
 
 
 
18
  history.append(messages[0])
19
 
 
 
 
20
  stream = client.chat.completions.create(
21
  model = model,
22
  messages = history,
@@ -35,7 +41,7 @@ app_text = gr.ChatInterface(
35
  fn = fn_text,
36
  type = "messages",
37
  additional_inputs = [
38
- #gr.Textbox(value="You are a helpful assistant.", label="System Prompt"),
39
  gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max Tokens"),
40
  gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
41
  gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-P"),
 
9
  def fn_text(
10
  prompt,
11
  history,
12
+ system_prompt,
13
  max_tokens,
14
  temperature,
15
  top_p,
16
  ):
17
+ #messages = [{"role": "user", "content": prompt}]
18
+ #history.append(messages[0])
19
+
20
+ messages = [{"role": "system", "content": system_prompt}]
21
  history.append(messages[0])
22
 
23
+ messages.append({"role": "user", "content": prompt})
24
+ history.append(messages[1])
25
+
26
  stream = client.chat.completions.create(
27
  model = model,
28
  messages = history,
 
41
  fn = fn_text,
42
  type = "messages",
43
  additional_inputs = [
44
+ gr.Textbox(value="You are a helpful assistant.", label="System Prompt"),
45
  gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max Tokens"),
46
  gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
47
  gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-P"),