Spaces:

sudip2003
/

Cat-Dog-classifier

Sleeping

sudip2003 commited on Jun 24

Commit

79b87e2

•

1 Parent(s): a9e59f3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ import numpy as np
 def load_models():
     models = {}
-    models['SimpleNN_model'] = tf.keras.models.load_model("SimpleNN_model.h5")
     models['VGG16'] = tf.keras.models.load_model("vgg16.h5")
     return models

 def load_models():
     models = {}
+    models['SimpleNN_model'] = tf.keras.models.load_model("Model_catsVSdogs.h5")
     models['VGG16'] = tf.keras.models.load_model("vgg16.h5")
     return models