Update app.py
Browse files
app.py
CHANGED
|
@@ -259,6 +259,10 @@ def check_password():
|
|
| 259 |
if not check_password():
|
| 260 |
st.stop()
|
| 261 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 262 |
dataset = load_dataset()
|
| 263 |
df = load_dataframe(dataset)
|
| 264 |
image_model = load_img_model()
|
|
|
|
| 259 |
if not check_password():
|
| 260 |
st.stop()
|
| 261 |
|
| 262 |
+
# Check if the directory exists
|
| 263 |
+
if not os.path.exists(model_dir):
|
| 264 |
+
download_models()
|
| 265 |
+
|
| 266 |
dataset = load_dataset()
|
| 267 |
df = load_dataframe(dataset)
|
| 268 |
image_model = load_img_model()
|