Update app.py

app.py

@@ -30,7 +30,7 @@ def replay(model_id, filename, environment, evaluate):
 	eval_env = VecFrameStack(eval_env, n_stack=4)
 
 	video_folder = 'logs/videos/'
-	video_length = 100
+	video_length = 1000
 
 
 	# Record the video starting at the first step
@@ -45,16 +45,16 @@ def replay(model_id, filename, environment, evaluate):
 	# Save the video
 	env.close()
 	
-	#mean_reward, std_reward = evaluate_policy(model, env, n_eval_episodes=10)
-	#print(f"mean_reward={mean_reward:.2f} +/- {std_reward}")
-	results = "hello" #f"mean_reward={mean_reward:.2f} +/- {std_reward}" 
+	mean_reward, std_reward = evaluate_policy(model, env, n_eval_episodes=10)
+	print(f"mean_reward={mean_reward:.2f} +/- {std_reward}")
+	results = f"mean_reward={mean_reward:.2f} +/- {std_reward}" 
 	print(type(results))
 	print(env)
 	print(env.video_recorder.path)
 	
 	videoclip = VideoFileClip(env.video_recorder.path)
 	videoclip.write_videofile("new_filename.mp4")
-	return 'new_filename.mp4'
+	return 'new_filename.mp4', results
 
 
 
@@ -66,5 +66,5 @@ iface = gr.Interface(fn=replay, inputs=[
                                         ]
 
 
-, outputs="video")
+, outputs=["video", "text"])
 iface.launch()