mohmehdi commited on
Commit
bce2308
·
verified ·
1 Parent(s): 1164f6b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,11 +4,11 @@ import gradio as gr
4
  client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
5
 
6
  def format_prompt(message, history):
7
- prompt = "<s>"
8
  # for user_prompt, bot_response in history:
9
  # prompt += f"[INST] {user_prompt} [/INST]"
10
  # prompt += f" {bot_response}</s> "
11
- prompt = f"[INST] {message} [/INST]"
12
  return prompt
13
 
14
  def generate(
 
4
  client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
5
 
6
  def format_prompt(message, history):
7
+ prompt = "<s>\n"
8
  # for user_prompt, bot_response in history:
9
  # prompt += f"[INST] {user_prompt} [/INST]"
10
  # prompt += f" {bot_response}</s> "
11
+ prompt = f"[INST] {message} [/INST]\n"
12
  return prompt
13
 
14
  def generate(