Thomas Simonini commited on
Commit
69db694
·
1 Parent(s): 579ad87

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -45,8 +45,9 @@ def replay(model_id, filename, environment, evaluate):
45
 
46
  mean_reward, std_reward = evaluate_policy(model, env, n_eval_episodes=10)
47
  print(f"mean_reward={mean_reward:.2f} +/- {std_reward}")
48
-
49
- return f"mean_reward={mean_reward:.2f} +/- {std_reward}"
 
50
 
51
 
52
  iface = gr.Interface(fn=replay, inputs=[
 
45
 
46
  mean_reward, std_reward = evaluate_policy(model, env, n_eval_episodes=10)
47
  print(f"mean_reward={mean_reward:.2f} +/- {std_reward}")
48
+ results = f"mean_reward={mean_reward:.2f} +/- {std_reward}"
49
+ print(type(results))
50
+ return results
51
 
52
 
53
  iface = gr.Interface(fn=replay, inputs=[