Upload 4 files

Dockerfile

@@ -17,4 +17,6 @@ EXPOSE 8501
 
 HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
 
-ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]
+RUN chmod +x run_app.sh
+
+ENTRYPOINT ["./run_app.sh"]

requirements.txt

@@ -1,3 +1,4 @@
 altair
 pandas
-streamlit
+streamlit
+mlipaudit @git+https://github.com/instadeepai/mlipaudit-open@main

run_app.sh

@@ -0,0 +1,22 @@
+#!/bin/bash
+
+HF_DATASET_REPO="InstaDeepAI/mlipaudit-results"
+HF_DATASET_LOCAL_PATH="./results"
+
+echo "Attempting to download dataset from ${HF_DATASET_REPO} to ${HF_DATASET_LOCAL_PATH}..."
+
+# Ensure the local data directory exists
+mkdir -p ${HF_DATASET_LOCAL_PATH}
+
+huggingface-cli download --repo-id ${HF_DATASET_REPO} --local-dir ${HF_DATASET_LOCAL_PATH} --repo-type dataset
+
+if [ $? -eq 0 ]; then
+    echo "Dataset downloaded successfully!"
+else
+    echo "Failed to download dataset."
+    exit 1 # Exit if download fails
+fi
+
+echo "Starting Streamlit app..."
+exec streamlit run src/app.py --server.port=8501 --server.address=0.0.0.0
+