Spaces:

ThomasSimonini
/

Stable-Baselines3

Sleeping

Thomas Simonini commited on Oct 5, 2021

Commit

cbdef19

•

1 Parent(s): db57f17

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ def replay(name):
   		return "./LunarLander-v2.mp4"
 	elif(option == "CartPole-v1 🕹️"):
   		return "./CartPole-v1.mp4"
-  	elif(option == "Atari Space Invaders 👾"):
   		return "./SpaceInvadersNoFrameskip-v4.mp4"
 """
@@ -33,7 +33,7 @@ iface = gr.Interface(
     [
         gr.inputs.Dropdown(["Atari Space Invaders 👾", "CartPole-v1 🕹️", "LunarLander-v2 🚀👩‍🚀 "]),
     ],
-    "playable_video"
 )

   		return "./LunarLander-v2.mp4"
 	elif(option == "CartPole-v1 🕹️"):
   		return "./CartPole-v1.mp4"
+	elif(option == "Atari Space Invaders 👾"):
   		return "./SpaceInvadersNoFrameskip-v4.mp4"
 """
     [
         gr.inputs.Dropdown(["Atari Space Invaders 👾", "CartPole-v1 🕹️", "LunarLander-v2 🚀👩‍🚀 "]),
     ],
+    "video"
 )