tried changing tensorflow device to cpu

app.py

@@ -14,6 +14,7 @@ st.write('To view how this model was trained go to the `Files and Versions` tab
 
 # Import Pre-trained Model
 model = tf.keras.models.load_model('mnist.h5')
+tf.device('/cpu:0')
 plt.rcParams.update({'font.size': 18})
 
 # Create a sidebar to hold the settings
@@ -62,6 +63,6 @@ if canvas_result.image_data is not None:
 
     # Show resized image
     with st.expander('Show Resized Image'):
-        st.write("The image needs to be resized, because it can only input 28x28 images")
+        st.write(
+            "The image needs to be resized, because it can only input 28x28 images")
         st.image(im, caption='Resized Image', width=28*9)
-