Spaces:
Paused
Paused
Update src/compliance_lib.py
Browse files- 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)
|