Thomas Simonini commited on
Commit
579ad87
·
1 Parent(s): 064e386

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -46,7 +46,7 @@ def replay(model_id, filename, environment, evaluate):
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}", "video"
50
 
51
 
52
  iface = gr.Interface(fn=replay, inputs=[
@@ -57,5 +57,5 @@ iface = gr.Interface(fn=replay, inputs=[
57
  ]
58
 
59
 
60
- , outputs=["text", "video"])
61
  iface.launch()
 
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=[
 
57
  ]
58
 
59
 
60
+ , outputs="text")
61
  iface.launch()