Update app.py

app.py

@@ -36,7 +36,7 @@ def replay(model_id, filename, environment, evaluate):
                        record_video_trigger=lambda x: x == 0, video_length=video_length,
                        name_prefix=f"test")
 
-	env.reset()
+	obs = env.reset()
 	for _ in range(video_length + 1):
 		action, _states = model.predict(obs)
 		obs, rewards, dones, info = env.step(action)