maulikanalog commited on
Commit
ec2848d
1 Parent(s): 0e87c57

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -25,7 +25,7 @@ def predict_bank_failure(balance_sheet_data, swift_model_id, sage_model_id, feed
25
  "api_config": api_configs[ENGINE],
26
  "temperature": float(swift_temperature),
27
  "top_p": float(swift_top_p),
28
- "max_tokens": 2048,
29
  }
30
 
31
  feedback_config = {
@@ -33,7 +33,7 @@ def predict_bank_failure(balance_sheet_data, swift_model_id, sage_model_id, feed
33
  "api_config": api_configs[ENGINE],
34
  "temperature": float(feedback_temperature),
35
  "top_p": float(feedback_top_p),
36
- "max_tokens": 2048,
37
  }
38
 
39
  sage_config = {
@@ -41,7 +41,7 @@ def predict_bank_failure(balance_sheet_data, swift_model_id, sage_model_id, feed
41
  "api_config": api_configs[ENGINE],
42
  "temperature": float(sage_temperature),
43
  "top_p": float(sage_top_p),
44
- "max_tokens": 2048,
45
  }
46
 
47
  # Try multiple locations for the prompt templates
 
25
  "api_config": api_configs[ENGINE],
26
  "temperature": float(swift_temperature),
27
  "top_p": float(swift_top_p),
28
+ "max_tokens": 4096,
29
  }
30
 
31
  feedback_config = {
 
33
  "api_config": api_configs[ENGINE],
34
  "temperature": float(feedback_temperature),
35
  "top_p": float(feedback_top_p),
36
+ "max_tokens": 4096,
37
  }
38
 
39
  sage_config = {
 
41
  "api_config": api_configs[ENGINE],
42
  "temperature": float(sage_temperature),
43
  "top_p": float(sage_top_p),
44
+ "max_tokens": 4096,
45
  }
46
 
47
  # Try multiple locations for the prompt templates