crazyup37 commited on
Commit
4639f07
1 Parent(s): a5db07f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ def format_prompt(message, custom_instructions=None):
15
  prompt += f"[INST] {message} [/INST]"
16
  return prompt
17
 
18
- def Mistral7B(prompt, instructions, api, temperature=0.1, max_new_tokens=2, top_p=0.95, repetition_penalty=1.0):
19
  global API_URL
20
  try:
21
  temperature = float(temperature)
 
15
  prompt += f"[INST] {message} [/INST]"
16
  return prompt
17
 
18
+ def Mistral7B(prompt, instructions, api, temperature=0.1, max_new_tokens=1, top_p=0.95, repetition_penalty=1.0):
19
  global API_URL
20
  try:
21
  temperature = float(temperature)