anktechsol commited on
Commit
43f4df8
·
verified ·
1 Parent(s): 51b780c

Update src/compliance_lib.py

Browse files
Files changed (1) hide show
  1. src/compliance_lib.py +1 -1
src/compliance_lib.py CHANGED
@@ -55,4 +55,4 @@ HF_MODEL = "mistralai/Mixtral-8x7B-Instruct-v0.1"
55
  def generate_report(prompt: str, max_tokens=600) -> str:
56
  client = InferenceClient(model=HF_MODEL, token=os.environ.get("HF_TOKEN"))
57
  return client.text_generation(prompt, max_new_tokens=max_tokens,
58
- temperature=0.4, top_p=0.9)
 
55
  def generate_report(prompt: str, max_tokens=600) -> str:
56
  client = InferenceClient(model=HF_MODEL, token=os.environ.get("HF_TOKEN"))
57
  return client.text_generation(prompt, max_new_tokens=max_tokens,
58
+ temperature=0.4, top_p=0.9)