rendyrg commited on
Commit
149e9bc
1 Parent(s): 0c0b6b9

change default to gpt4 and fix object ref possible bug

Browse files
Files changed (1) hide show
  1. app.py +7 -8
app.py CHANGED
@@ -2,27 +2,26 @@ import gradio as gr
2
  import openai
3
 
4
 
5
- def get_answer_chatgpt(chat_history, message, system_message=None, model="gpt-3.5-turbo", max_length=1000):
6
  messages = []
7
  if model is None:
8
- model = "gpt-3.5-turbo"
9
 
10
  if system_message or system_message != "":
11
  TEMPLATE_SYSTEM = {"role": "system", "content": system_message}
12
  messages.append(TEMPLATE_SYSTEM)
13
 
14
- TEMPLATE_USER = {"role": "user"}
15
- TEMPLATE_BOT = {"role": "assistant"}
16
-
17
  # add historical conversation
18
  for pair in chat_history:
19
- TEMPLATE_USER["content"] = pair[0]
20
- TEMPLATE_BOT["content"] = pair[1]
21
  messages.append(TEMPLATE_USER)
22
  messages.append(TEMPLATE_BOT)
 
23
  # add latest input
24
  TEMPLATE_USER["content"] = message
25
  messages.append(TEMPLATE_USER)
 
26
  response = openai.ChatCompletion.create(
27
  model=model,
28
  messages=messages,
@@ -46,7 +45,7 @@ with gr.Blocks() as demo:
46
  clear = gr.Button("Clear all converstation")
47
  with gr.Column(scale=5):
48
  with gr.Accordion("Parameters", open=False):
49
- model = gr.Dropdown(["gpt-3.5-turbo", "gpt-4"], label="Model", value="gpt-3.5-turbo")
50
  max_length = gr.Slider(0, 10000, value=1000, step=1, label="Maximum length token", interactive=True)
51
 
52
  system_message = gr.Textbox(placeholder="You are helpful assistence", lines=5, label="System Message")
 
2
  import openai
3
 
4
 
5
+ def get_answer_chatgpt(chat_history, message, system_message=None, model="gpt-4", max_length=1000):
6
  messages = []
7
  if model is None:
8
+ model = "gpt-4"
9
 
10
  if system_message or system_message != "":
11
  TEMPLATE_SYSTEM = {"role": "system", "content": system_message}
12
  messages.append(TEMPLATE_SYSTEM)
13
 
 
 
 
14
  # add historical conversation
15
  for pair in chat_history:
16
+ TEMPLATE_USER = {"role": "user", "content": pair[0]}
17
+ TEMPLATE_BOT = {"role": "assistant", "content": pair[1]}
18
  messages.append(TEMPLATE_USER)
19
  messages.append(TEMPLATE_BOT)
20
+
21
  # add latest input
22
  TEMPLATE_USER["content"] = message
23
  messages.append(TEMPLATE_USER)
24
+
25
  response = openai.ChatCompletion.create(
26
  model=model,
27
  messages=messages,
 
45
  clear = gr.Button("Clear all converstation")
46
  with gr.Column(scale=5):
47
  with gr.Accordion("Parameters", open=False):
48
+ model = gr.Dropdown(["gpt-4", "gpt-3.5-turbo"], label="Model", value="gpt-4")
49
  max_length = gr.Slider(0, 10000, value=1000, step=1, label="Maximum length token", interactive=True)
50
 
51
  system_message = gr.Textbox(placeholder="You are helpful assistence", lines=5, label="System Message")