Update app.py

app.py

@@ -12,6 +12,8 @@ from stable_baselines3.common.vec_env import VecFrameStack
 from stable_baselines3.common.vec_env import VecVideoRecorder, DummyVecEnv
 from stable_baselines3.common.evaluation import evaluate_policy
 
+from moviepy.editor import *
+
 def replay(model_id, filename, environment, evaluate):
 	# Load the model
 	checkpoint = load_from_hub(model_id, filename)
@@ -49,7 +51,11 @@ def replay(model_id, filename, environment, evaluate):
 	print(type(results))
 	print(env)
 	print(env.video_recorder.path)
-	return env.video_recorder.path
+	
+	videoclip = VideoFileClip(env.video_recorder.path)
+	videoclip.write_videofile("new_filename.mp4")
+	return 'new_filename.mp4'
+
 
 
 iface = gr.Interface(fn=replay, inputs=[