latex add caption to equation code example

Example 1: latex image caption

\begin{figure}
    \includegraphics[width=\textwidth]{Picture}
    \caption{Picture caption}
\end{figure}

Example 2: equation with captionlatex

\begin{figure}
\[ E = m c^2 \]
\caption{A famous equation}
\end{figure}