Update app.py

app.py

@@ -17,6 +17,8 @@ models = {'VGG16': 'vgg16', 'VGG19': 'vgg16', 'ResNet50V2': 'resnet_v2',
 model_type2 = models[model_type]
 
 top_n = st.sidebar.selectbox('Number of Results', (3, 5, 10))
+results = st.sidebar.selectbox('Display Summary', ('No','Yes'))
+
 
 exec(f'from keras.applications.{model_type2} import {model_type}')
 exec(
@@ -47,3 +49,10 @@ df['Percent Certainty'] = df['Percent Certainty'].apply(
     lambda x: '{:.2%}'.format(x))
 
 st.dataframe(df)
+
+if results=='Yes':
+    stringlist = []
+    model.summary(print_fn=lambda x: stringlist.append(x))
+    short_model_summary = "\n".join(stringlist)
+    print(short_model_summary)
+    st.write(short_model_summary)