Spaces:

gabri14el
/

grapevine_classicationv2

Sleeping

gabri14el commited on May 22, 2023

Commit

efaba4e

1 Parent(s): c10cf42

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -53,7 +53,7 @@ def define_model(model):
     weights = get_weights(model)
     if model == 'convnext':
         preprocessing_function=tf.keras.applications.convnext.preprocess_input
-        model = tf.keras.applications.ConvNeXtLarge(
             include_top=False,
             input_shape= (target_size_dimension, target_size_dimension, 3),
             weights='imagenet',

     weights = get_weights(model)
     if model == 'convnext':
         preprocessing_function=tf.keras.applications.convnext.preprocess_input
+        model = tf.keras.applications.ConvNeXtBase(
             include_top=False,
             input_shape= (target_size_dimension, target_size_dimension, 3),
             weights='imagenet',