ArmelR commited on
Commit
943046d
·
1 Parent(s): 130d506

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -98,14 +98,14 @@ def get_inference_prompt(messages, model_name):
98
  if message["role"] == "user" :
99
  prompt += f"<|user|>\n{message['content']}<|endoftext|>\n<|assistant|>"
100
  else : # message["role"] == "assistant"
101
- prompt += f"{message['content']}<|endoftext|>\n"
102
  elif model_name == "starChat-alpha" :
103
  prompt = "<|system|>\n<|end|>\n"
104
  for message in messages :
105
  if message["role"] == "user" :
106
  prompt += f"<|user|>\n{message['content']}<|end|>\n<|assistant|>"
107
  else : # message["role"] == "assistant"
108
- prompt += f"{message['content']}<|end|>\n"
109
  else : # starCoder-gradio
110
  prompt = ""
111
  for message in messages :
 
98
  if message["role"] == "user" :
99
  prompt += f"<|user|>\n{message['content']}<|endoftext|>\n<|assistant|>"
100
  else : # message["role"] == "assistant"
101
+ prompt += f"\n{message['content']}<|endoftext|>\n"
102
  elif model_name == "starChat-alpha" :
103
  prompt = "<|system|>\n<|end|>\n"
104
  for message in messages :
105
  if message["role"] == "user" :
106
  prompt += f"<|user|>\n{message['content']}<|end|>\n<|assistant|>"
107
  else : # message["role"] == "assistant"
108
+ prompt += f"\n{message['content']}<|end|>\n"
109
  else : # starCoder-gradio
110
  prompt = ""
111
  for message in messages :