Update app.py

- Remove nb of episodes

app.py

@@ -32,13 +32,13 @@ def load_model(env_name):
 
     return model
 
-def replay(env_name, time_sleep, num_episodes):
+def replay(env_name, time_sleep): #, num_episodes):
   env = load_env(env_name)
   model = load_model(env_name)
-  for i in range(num_episodes):
-    obs = env.reset()
-    done = False
-    while not done:
+  #for i in range(num_episodes):
+  obs = env.reset()
+  done = False
+  while not done:
       frame = env.render(mode="rgb_array")
       action, _states = model.predict(obs)
       obs, reward, done, info = env.step([action])
@@ -53,11 +53,11 @@ demo = gr.Interface(
         "SeaquestNoFrameskip-v4",
         "QbertNoFrameskip-v4",
         ]),
-     gr.Slider(0.01, 1, value=0.1),
-     gr.Slider(1, 20, value=5)
+     gr.Slider(0.01, 1, value=0.05),
+     #gr.Slider(1, 20, value=5)
      ],
     gr.Image(),
     title="Watch Agents playing Atari games 🤖",
     description="Select an environment to watch a Hugging Face's trained deep reinforcement learning agent.",
-    article = "time_sleep is the time delay between each frame (0.1 by default)."
-).launch(debug=True, enable_queue=True).queue()
+    article = "time_sleep is the time delay between each frame (0.05 by default)."
+).launch().queue()