Update app.py

app.py

@@ -47,6 +47,8 @@ def initialize_simulator(image_name):
 
 @spaces.GPU
 def model(direction):
+    global cached_latent_frames
+    global cached_actions
     if direction == 'right':
         action = np.array([0, 0.05])
     elif direction == 'left':
@@ -59,8 +61,7 @@ def model(direction):
         raise ValueError(f"Invalid direction: {direction}")
     genie_result = genie.step(action,  cached_latent_frames=cached_latent_frames, cached_actions=cached_actions)
     next_image = genie_result['pred_next_frame']
-    global cached_latent_frames
-    global cached_actions
+
     cached_latent_frames = genie_result['set_cached_latent_frames']
     cached_actions = genie_result['set_cached_actions']
     next_image = cv2.resize(next_image, (RES, RES))