iamsubrata commited on
Commit
ee452fc
1 Parent(s): 40b5068

Update app/app.py

Browse files

Display the preprocessing steps

Files changed (1) hide show
  1. app/app.py +5 -0
app/app.py CHANGED
@@ -47,6 +47,9 @@ def app():
47
  with open("models/unfreezed_model_summary.pkl", "rb") as f:
48
  unfreezed_arch_summary = pickle.load(f)
49
 
 
 
 
50
  #######################################################################################################################
51
 
52
  if selected == "Upload":
@@ -64,6 +67,8 @@ def app():
64
 
65
  if selected == "Model":
66
  model_info()
 
 
67
  st.subheader("FastAi Model Summary (Freezed)")
68
  st.code(freezed_arch_summary)
69
  st.subheader("FastAi Model Summary (Unfreezed)")
 
47
  with open("models/unfreezed_model_summary.pkl", "rb") as f:
48
  unfreezed_arch_summary = pickle.load(f)
49
 
50
+ with open("models/birds_model_preprocessing.pkl","rb") as f:
51
+ preprocessing_steps = pickle.load(f)
52
+
53
  #######################################################################################################################
54
 
55
  if selected == "Upload":
 
67
 
68
  if selected == "Model":
69
  model_info()
70
+ st.subheader("Preprocessing Steps")
71
+ st.code(preprocessing_steps)
72
  st.subheader("FastAi Model Summary (Freezed)")
73
  st.code(freezed_arch_summary)
74
  st.subheader("FastAi Model Summary (Unfreezed)")