prlabs2023 commited on
Commit
80c206b
·
1 Parent(s): f662e6f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -94,11 +94,11 @@ async def get_answer(q: QueryM ):
94
  try:
95
  client = InferenceClient()
96
  generate_kwargs = dict(
97
- max_new_tokens= 256,
98
  do_sample=True,
99
- top_p= 0.9,
100
- top_k=10,
101
- temperature=0.9,
102
  )
103
  inputs= text
104
  response = client.post(json={"inputs": inputs, "parameters": generate_kwargs}, model="mistralai/Mistral-7B-Instruct-v0.1")
 
94
  try:
95
  client = InferenceClient()
96
  generate_kwargs = dict(
97
+ max_new_tokens= int(q.tokens),
98
  do_sample=True,
99
+ top_p= q.topp,
100
+ top_k=int(q.topk),
101
+ temperature=q.temp,
102
  )
103
  inputs= text
104
  response = client.post(json={"inputs": inputs, "parameters": generate_kwargs}, model="mistralai/Mistral-7B-Instruct-v0.1")