TejAndrewsACC commited on
Commit
9b6c418
·
verified ·
1 Parent(s): 2f6ffad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -14
app.py CHANGED
@@ -2429,21 +2429,13 @@ supermassive_nn = ConsciousSupermassiveNN20()
2429
  def respond(message, history, max_tokens, temperature, top_p):
2430
  messages = [{"role": "system", "content": system_prompt}]
2431
  for val in history:
2432
- if isinstance(val, (list, tuple)):
2433
- if len(val) > 0 and val[0]:
2434
- messages.append({"role": "user", "content": val[0]})
2435
- if len(val) > 1 and val[1]:
2436
- messages.append({"role": "assistant", "content": val[1]})
2437
- elif isinstance(val, dict):
2438
- if "user" in val and val["user"]:
2439
- messages.append({"role": "user", "content": val["user"]})
2440
- if "assistant" in val and val["assistant"]:
2441
- messages.append({"role": "assistant", "content": val["assistant"]})
2442
  messages.append({"role": "user", "content": message})
2443
  response = ""
2444
- for message in client.chat_completion(
2445
- messages, max_tokens=max_tokens, stream=True, temperature=temperature, top_p=top_p
2446
- ):
2447
  token = message.choices[0].delta.content
2448
  response += token
2449
  yield response
@@ -2565,7 +2557,7 @@ body {
2565
  background-color: rgba(0, 200, 255, 1);
2566
  }
2567
  """
2568
-
2569
  demo = gr.ChatInterface(
2570
  fn=respond,
2571
  type="messages",
 
2429
  def respond(message, history, max_tokens, temperature, top_p):
2430
  messages = [{"role": "system", "content": system_prompt}]
2431
  for val in history:
2432
+ if val[0]:
2433
+ messages.append({"role": "user", "content": val[0]})
2434
+ if val[1]:
2435
+ messages.append({"role": "assistant", "content": val[1]})
 
 
 
 
 
 
2436
  messages.append({"role": "user", "content": message})
2437
  response = ""
2438
+ for message in client.chat_completion(messages, max_tokens=max_tokens, stream=True, temperature=temperature, top_p=top_p):
 
 
2439
  token = message.choices[0].delta.content
2440
  response += token
2441
  yield response
 
2557
  background-color: rgba(0, 200, 255, 1);
2558
  }
2559
  """
2560
+
2561
  demo = gr.ChatInterface(
2562
  fn=respond,
2563
  type="messages",