Spaces:

Serrelab
/

fossil_app

Sleeping

andy-wyx commited on Jun 11, 2024

Commit

1c118fd

1 Parent(s): df27250

add compare resnet model

Files changed (1) hide show

app.py CHANGED Viewed

@@ -120,7 +120,7 @@ def get_model(model_name):
                                   embedding_units = 256,
                                   embedding_depth = 2,
                                   n_classes = n_classes)
-        model.load_weights('model_classification/fossil-model.h5')
     elif model_name == 'Fossils':
         n_classes = 142
         model,_,_ = get_resnet_model('model_classification/fossil-model.h5')

                                   embedding_units = 256,
                                   embedding_depth = 2,
                                   n_classes = n_classes)
+        model.load_weights('model_classification/fossil-new.h5')
     elif model_name == 'Fossils':
         n_classes = 142
         model,_,_ = get_resnet_model('model_classification/fossil-model.h5')