final

app.py

@@ -30,7 +30,7 @@ def import_model(model_name):
 # Function to predict outputs
 def predict(img, model_name):
     if (model_name == 'Geometric Figures & Solids'):
-        geo_learn = import_model('geo_model.pkl')
+        geo_learn = import_model('geometric_model.pkl')
         preds = geo_learn.predict(img)
     elif (model_name == 'Scenery, Natural Phenomena'):
         landscape_learn = import_model('landscape_model.pkl')
@@ -42,7 +42,7 @@ def predict(img, model_name):
         colors_learn = import_model('colors_model.pkl')
         preds = colors_learn.predict(img) 
     elif (model_name == 'Buildings, Dwellings & Furniture'):
-        build_learn = import_model('buildings_model.pkl')
+        build_learn = import_model('buildings.pkl')
         preds = build_learn.predict(img)
     elif (model_name == 'Animals'):
         anim_learn = import_model('animals.pkl')