Soft_max library

Hi I’m trying to code the practice lab at the end on Google Colab and having trouble importing libraries. I have the files int he right drive but I keep getting an error that library not found. Please help!

from google.colab import drive
drive.mount(‘/content/gdrive/’, force_remount=True)
%cd “/content/gdrive/MyDrive/python/C2_W1”
import numpy as np
import tensorflow as tf
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Dense
import matplotlib.pyplot as plt
%matplotlib inline
from autils import *
from public_tests import *
import logging
#getting error, not found
from lab_utils_softmax import plt_softmax

ModuleNotFoundError Traceback (most recent call last)
in <cell line: 14>()
12 tf.autograph.set_verbosity(0)
13 ##getting error, not found
—> 14 from lab_utils_softmax import plt_softmax
15 np.set_printoptions(precision=2)

ModuleNotFoundError: No module named ‘lab_utils_softmax’

Please check this guide.