flag update

app.py

@@ -4,9 +4,10 @@ import numpy as np # linear algebra
 import os #interacting with input and output directories
 import tensorflow as tf #framework for creating the neural network
 
+model = tf.keras.models.load_model(os.path.join(os.getcwd(), 'noteClassifierModel.h5'))
+
 def fn(img):
     img = np.expand_dims(img, axis = 0)
-    model = tf.keras.models.load_model(os.path.join(os.getcwd(), 'noteClassifierModel.h5'))
     pred = model.predict(img)
     pred = np.argmax(pred)
 
@@ -29,6 +30,7 @@ iface = gr.Interface(fn,
                      title="Note Classifier",
                      description=description,
                      theme="dark-seafoam",
-                     allow_flagging="manual")
+                     allow_flagging="manual",
+                     flagging_dir='flagging records')
 iface.launch(inline=False)