Hemang1915 commited on
Commit
1741953
·
1 Parent(s): a791c56

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -120,7 +120,7 @@ async def predict(request: TransactionRequest):
120
  if request.use_rl:
121
  ppo = load_ppo_model()
122
  obs = pad[0]
123
- action, _ = ppo.predict(obs)
124
  print(f"RL Action: {action}, Observation: {obs}")
125
  cat_idx = action // num_subcategories
126
  subcat_idx = action % num_subcategories
 
120
  if request.use_rl:
121
  ppo = load_ppo_model()
122
  obs = pad[0]
123
+ action, _ = ppo.predict(obs, deterministic=True) # Force deterministic
124
  print(f"RL Action: {action}, Observation: {obs}")
125
  cat_idx = action // num_subcategories
126
  subcat_idx = action % num_subcategories