iamsubrata
commited on
Commit
•
ee452fc
1
Parent(s):
40b5068
Update app/app.py
Browse filesDisplay the preprocessing steps
- 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)")
|