annikwag commited on
Commit
f2efbb4
·
verified ·
1 Parent(s): 67fdfe6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -39,7 +39,7 @@ def get_rag_answer(query, top_results):
39
  headers = {"Authorization": f"Bearer {WRITE_ACCESS_TOKEN}"}
40
  payload = {
41
  "inputs": prompt,
42
- "parameters": {"max_new_tokens": 500}
43
  }
44
  response = requests.post(DEDICATED_ENDPOINT, headers=headers, json=payload)
45
  if response.status_code == 200:
 
39
  headers = {"Authorization": f"Bearer {WRITE_ACCESS_TOKEN}"}
40
  payload = {
41
  "inputs": prompt,
42
+ "parameters": {"max_new_tokens": 220}
43
  }
44
  response = requests.post(DEDICATED_ENDPOINT, headers=headers, json=payload)
45
  if response.status_code == 200: