add support for categories.txt

app.py

@@ -13,11 +13,22 @@ import h5py # remove later
 ###           model loading          ###
 ### -------------------------------- ###
 
-model = load_model('model.h5') # single file model
-labels = ['cheetah', 'hyena', 'jaguar', 'tiger'] # will need to be loaded from file in the order of your directories
+model = load_model('model.h5') # single file model from colab
+labels = ['please upload categories.txt' for i in range(100)] # placeholder
 
-## -------------------------------- ###
-###            file reading          ###
+
+## --------------------------------- ###
+###     reading: categories.txt      ###
+### -------------------------------- ###
+if os.path.isfile("categories.txt"):
+    # open info.txt in read mode
+    categories = open("categories.txt", "r")
+    labels = categories.readline().split()
+    print(labels)
+
+
+## --------------------------------- ###
+###        reading: info.txt         ###
 ### -------------------------------- ###
 # placeholders in case info.txt does not exist
 placeholder =  "please create an info.txt to customize this text"