Update classification.py

classification.py

@@ -6,7 +6,7 @@ import tensorflow as tf
 import streamlit as st
 # with tf.device('/cpu:0'):
 # Load the saved model
-model = tf.keras.models.load_model('./best_resnet152_model.h5')
+model = tf.keras.models.load_model('best_resnet152_model.h5')
 
 class_names = {0: '1099_Div', 1: '1099_Int', 2: 'Non_Form', 3: 'w_2', 4: 'w_3'}
 # print(class_names)