fix

app.py

@@ -51,14 +51,23 @@ def predict(img, model_name):
         colors_learn = import_model('colors_model.pkl')
         pred = colors_learn.predict(img)[0]
         labels = misc
+        pred = str(pred)
+        fix_pred = pred[4:-4]
+        return fix_pred
     elif (model_name == 'Buildings, Dwellings & Furniture'):
         build_learn = import_model('buildings_model.pkl')
         pred = build_learn.predict(img)[0]
         labels = build
+        pred = str(pred)
+        fix_pred = pred[4:-4]
+        return fix_pred
     elif (model_name == 'Animals'):
         anim_learn = import_model('animals_model.pkl')
         pred = anim_learn.predict(img)[0]
         labels = anim
+        pred = str(pred)
+        fix_pred = pred[4:-4]
+        return fix_pred
 
     pred = str(pred)
     fix_pred = pred[4:-4]