CodingBuddy commited on
Commit
77cb43b
·
verified ·
1 Parent(s): 87a10ec

Upload folder using huggingface_hub

Browse files
Files changed (4) hide show
  1. Dockerfile +15 -12
  2. app.py +62 -0
  3. config.py +2 -0
  4. requirements.txt +7 -3
Dockerfile CHANGED
@@ -1,20 +1,23 @@
1
- FROM python:3.13.5-slim
 
2
 
 
3
  WORKDIR /app
4
 
5
- RUN apt-get update && apt-get install -y \
6
- build-essential \
7
- curl \
8
- git \
9
- && rm -rf /var/lib/apt/lists/*
10
-
11
- COPY requirements.txt ./
12
- COPY src/ ./src/
13
 
 
14
  RUN pip3 install -r requirements.txt
15
 
16
- EXPOSE 8501
 
 
 
 
 
17
 
18
- HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
19
 
20
- ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]
 
 
1
+ # Use a minimal base image with Python 3.9 installed
2
+ FROM python:3.9
3
 
4
+ # Set the working directory inside the container to /app
5
  WORKDIR /app
6
 
7
+ # Copy all files from the current directory on the host to the container's /app directory
8
+ COPY . .
 
 
 
 
 
 
9
 
10
+ # Install Python dependencies listed in requirements.txt
11
  RUN pip3 install -r requirements.txt
12
 
13
+ RUN useradd -m -u 1000 user
14
+ USER user
15
+ ENV HOME=/home/user \
16
+ PATH=/home/user/.local/bin:$PATH
17
+
18
+ WORKDIR $HOME/app
19
 
20
+ COPY --chown=user . $HOME/app
21
 
22
+ # Define the command to run the Streamlit app on port "8501" and make it accessible externally
23
+ CMD ["streamlit", "run", "app.py", "--server.port=8501", "--server.address=0.0.0.0", "--server.enableXsrfProtection=false"]
app.py ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ import streamlit as st
3
+ import pandas as pd
4
+ from huggingface_hub import hf_hub_download
5
+ import joblib
6
+
7
+ from config import HF_REPO_ID
8
+
9
+ # Download and load the model
10
+ model_path = hf_hub_download(repo_id=HF_REPO_ID, filename="best_tourism_package_prediction_model_v1.joblib")
11
+ model = joblib.load(model_path)
12
+
13
+ # Streamlit UI for Tourism Package Prediction...
14
+ st.title("Tour Package Prediction App")
15
+ st.write("""
16
+ This application predicts the likelihood of a customer selecting the package parameters.
17
+ Please enter the sensor and configuration data below to get a prediction.
18
+ """)
19
+
20
+ # User input
21
+ st.header("User Input")
22
+
23
+ Age = st.number_input("Age", min_value=10, max_value=100, value=30, step=1)
24
+ TypeofContact = st.selectbox("TypeofContact", ["Self Enquiry", "Company Invited"])
25
+ CityTier = st.selectbox("CityTier", ["Tier 1", "Tier 2", "Tier 3"])
26
+ Occupation = st.selectbox("Occupation", ["Salaried", "Freelancer"])
27
+ Gender = st.selectbox("Gender", ["Male", "Female"])
28
+ NumberOfPersonVisiting = st.number_input("Number of person visiting", min_value=1, max_value=10, value=2, step=1)
29
+ PreferredPropertyStar = st.number_input("Preferred Property Star", min_value=2, max_value=5, value=3, step=1)
30
+ MaritalStatus = st.selectbox("Marital Status", ["Single", "Married", "Divorced"])
31
+ NumberOfTrips = st.number_input("Number of trips", min_value=1, max_value=10, value=2, step=1)
32
+ Passport = st.selectbox("Passport", ["Yes", "No"])
33
+ OwnCar = st.selectbox("Own Car", ["Yes", "No"])
34
+ NumberOfChildrenVisiting = st.number_input("Number of children visiting", min_value=0, max_value=5, value=0, step=1)
35
+ Designation = st.selectbox("Designation", ["Manager", "Executive", "Senior Manager", "VP"])
36
+ MonthlyIncome = st.number_input("Monthly Income", min_value=0, max_value=100000, value=50000, step=100)
37
+
38
+
39
+ # Assemble input into DataFrame
40
+ input_data = pd.DataFrame([{
41
+ 'Age': Age,
42
+ 'TypeofContact': TypeofContact,
43
+ 'CityTier': CityTier,
44
+ 'Occupation': Occupation,
45
+ 'Gender': Gender,
46
+ 'NumberOfPersonVisiting': NumberOfPersonVisiting,
47
+ 'PreferredPropertyStar': PreferredPropertyStar,
48
+ 'MaritalStatus': MaritalStatus,
49
+ 'NumberOfTrips': NumberOfTrips,
50
+ 'Passport': Passport,
51
+ 'OwnCar': OwnCar,
52
+ 'NumberOfChildrenVisiting': NumberOfChildrenVisiting,
53
+ 'Designation': Designation,
54
+ 'MonthlyIncome': MonthlyIncome
55
+ }])
56
+
57
+
58
+ if st.button("Predict Failure"):
59
+ prediction = model.predict(input_data)[0]
60
+ result = "Package selected" if prediction == 1 else "No Failure"
61
+ st.subheader("Prediction Result:")
62
+ st.success(f"The model predicts: **{result}**")
config.py ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ # common configurations and constants
2
+ HF_REPO_ID = "CodingBuddy/Tourism_package_prediction"
requirements.txt CHANGED
@@ -1,3 +1,7 @@
1
- altair
2
- pandas
3
- streamlit
 
 
 
 
 
1
+ pandas==2.2.2
2
+ huggingface_hub==0.32.6
3
+ streamlit==1.43.2
4
+ joblib==1.5.1
5
+ scikit-learn==1.6.0
6
+ xgboost==2.1.4
7
+ mlflow==3.0.1