debisoft commited on
Commit
9b86a92
1 Parent(s): 33f3a35
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ def score_model(model_uri, databricks_token, prompt):
15
  dataset=pd.DataFrame({
16
  "prompt":[prompt],
17
  "temperature": [0.5],
18
- "max_tokens": [1500]})
19
  headers = {
20
  "Authorization": f"Bearer {databricks_token}",
21
  "Content-Type": "application/json",
 
15
  dataset=pd.DataFrame({
16
  "prompt":[prompt],
17
  "temperature": [0.5],
18
+ "max_tokens": [500]})
19
  headers = {
20
  "Authorization": f"Bearer {databricks_token}",
21
  "Content-Type": "application/json",