add import for os

app.py

@@ -5,6 +5,7 @@
 from tensorflow.keras.models import load_model
 import gradio as gr # remove later
 import numpy as np
+import os
 from yattag import Doc
 import h5py # remove later