Update main.py

main.py

@@ -122,7 +122,7 @@ def predict_next_frame(previous_frames: List[np.ndarray], previous_actions: List
     
     # Convert the generated frame to the correct format
     new_frame = new_frame.transpose(1, 2, 0)
-        print (new_frame.max(), new_frame.min())
+    print (new_frame.max(), new_frame.min())
     new_frame_denormalized = denormalize_image(new_frame, source_range=(0, 1))
     
     # Draw the trace of previous actions