Upload folder using huggingface_hub

Dockerfile

@@ -10,7 +10,7 @@ COPY . .
 # Install Python dependencies listed in requirements.txt
 RUN pip3 install -r requirements.txt
 
-# Define the command to run the Streamlit app on port 8501 and make it accessible externally
-CMD ["streamlit", "run", "app.py", "--server.port=8501", "--server.address=0.0.0.0", "--server.enableXsrfProtection=false"]
+# Define the command to run the Streamlit app on port 7860 and make it accessible externally
+CMD ["streamlit", "run", "app.py", "--server.port=7860", "--server.enableCORS=false", "--server.address=0.0.0.0", "--server.enableXsrfProtection=false"]
 
 # NOTE: Disable XSRF protection for easier external access in order to make batch predictions

app.py

@@ -1,10 +1,16 @@
 import streamlit as st
 import pandas as pd
 import requests
+import time
 
 # Set the title of the Streamlit app
 st.title("SuperKart Sales Prediction")
 
+# Health check endpoint for Hugging Face
+if st.query_params.get("healthcheck") == "true":
+    st.write("OK")
+    st.stop()
+
 # Section for single prediction
 st.subheader("Single Product-Store Prediction")
 

requirements.txt

@@ -1,3 +1,4 @@
 pandas==2.2.2
 requests==2.28.1
 streamlit==1.43.2
+numpy==2.0.2