Update app.py

app.py

@@ -33,7 +33,7 @@ def load_model(env_name):
     model = PPO.load(checkpoint, custom_objects=custom_objects)
 
     return model
-
+    
 def replay(env_name, max_steps, time_sleep):
   env = load_env(env_name)
   model = load_model(env_name)
@@ -42,15 +42,15 @@ def replay(env_name, max_steps, time_sleep):
   done = False
   i = 0
   while not done:
-      i+= 1
-      if i < max_steps:
-          frame = env.render(mode="rgb_array")
-          action, _states = model.predict(obs)
-          obs, reward, done, info = env.step([action])
-          time.sleep(time_sleep)
-          yield frame
-     else:
-         break
+    i+= 1
+    if i < max_steps:
+      frame = env.render(mode="rgb_array")
+      action, _states = model.predict(obs)
+      obs, reward, done, info = env.step([action])
+      time.sleep(time_sleep)
+      yield frame
+    else:
+      break
 
 demo = gr.Interface(
     replay,