Spaces:

mattritchey
/

Keras_Image_Classifier

Sleeping

mattritchey commited on Apr 21, 2023

Commit

f577116

1 Parent(s): d2fc011

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,7 +22,7 @@ st.set_page_config(layout="wide")
 model_type = st.sidebar.selectbox(
     'Select Model', ('VGG16', 'VGG19', 'ResNet50V2', 'MobileNetV2'))
-models = {'VGG16': 'vgg16', 'VGG19': 'vgg16', 'ResNet50V2': 'resnet_v2',
           'MobileNetV2': 'mobilenet_v2'}
 model_type2 = models[model_type]

 model_type = st.sidebar.selectbox(
     'Select Model', ('VGG16', 'VGG19', 'ResNet50V2', 'MobileNetV2'))
+models = {'VGG16': 'vgg16', 'VGG19': 'vgg19', 'ResNet50V2': 'resnet_v2',
           'MobileNetV2': 'mobilenet_v2'}
 model_type2 = models[model_type]