salomonsky commited on
Commit
55b59a8
·
verified ·
1 Parent(s): 70225f6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -43,7 +43,7 @@ def format_prompt(message, history):
43
  prompt += f"[INST] {message} [/INST]"
44
  return prompt
45
 
46
- def generate(audio_text, history, temperature=None, max_new_tokens=512, top_p=0.95, repetition_penalty=1.0):
47
  client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
48
 
49
  temperature = float(temperature) if temperature is not None else 0.9
 
43
  prompt += f"[INST] {message} [/INST]"
44
  return prompt
45
 
46
+ def generate(audio_text, history, temperature=None, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0):
47
  client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
48
 
49
  temperature = float(temperature) if temperature is not None else 0.9