Upload folder using huggingface_hub

Dockerfile

@@ -1,23 +1,12 @@
-FROM ubuntu:22.04
 
-WORKDIR /app
+FROM python:3.9-slim
 
-RUN apt-get update && apt-get install -y \
-    build-essential \
-    curl \
-    software-properties-common \
-    git \
-    && rm -rf /var/lib/apt/lists/*
+WORKDIR /app
 
-COPY requirements.txt ./
-COPY src/ ./src/
+COPY ./app
 
-RUN apt-get update && \
-    apt-get install -y python3 python3-pip && \
-    pip3 install -r requirements.txt
+RUN pip3 install -r requirements.txt
 
 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"]
+CMD ["streamlit","run","app.py","--server.port=8501","--server.address=0.0.0.0","--server.enableXsrfProtection=false"]

app.py

@@ -0,0 +1,61 @@
+
+import pandas as pd
+import numpy as np
+import requests
+import streamlit as st
+
+st.title('Customer Churn Prediction Frontend')
+
+st.subheader('Online Prediction')
+
+CustomerId=st.number_input('CustomerId',min_value=10000,max_value=99999999)
+Surname=st.text_input('Surname')
+CreditScore=st.number_input('CreditScore',min_value=300,max_value=900,value=450)
+Geography=st.selectbox('Geography',['France','Spain','Germany'])
+Age=st.number_input('Age',min_value=18,max_value=100,value=30)
+Tenure=st.number_input('Tenure',min_value=0,max_value=10,value=5)
+Balance=st.number_input('Balance',min_value=0.00,max_value=99999.99,value=97198.54)
+NumOfProducts=st.number_input('NumOfProducts',min_value=1,max_value=4,value=2)
+HasCrCard=st.selectbox('HasCrCard',['Yes','No'])
+IsActiveMember=st.selectbox('IsActiveMember',['Yes','No'])
+EstimatedSalary=st.number_input('EstimatedSalary',min_value=10.00,max_value=999999.99)
+
+input_data={'CustomerId':CustomerId,
+            'Surname':Surname,
+            'CreditScore':CreditScore,
+            'Geography':Geography,
+            'Age':Age,
+            'Tenure':Tenure,
+            'Balance':Balance,
+            'NumOfProducts':NumOfProducts,
+            'HasCrCard':1 if HasCrCard=='Yes' else 0,
+            'IsActiveMember':1 if IsActiveMember=='Yes' else 0,
+            'EstimatedSalary':EstimatedSalary
+             }
+
+if st.button('Predict'):
+  response=requests.post("https://siddhesh1981-CustomerChurnBackend.hf.space/Predict/Data",json=input_data)
+  if response.status_code==200:
+    result=response.json()
+    st.success(f"Based on the given input information the Customer with id {CustomerId} and Surname {Surname} is expected to {result['predict_label']}")
+  else:
+    st.error(response.text)
+
+st.subheader('Batch Prediction')
+
+file2=st.file_uploader('Upload a csv file',type=['csv'])
+if file2 is not None:
+  if st.button('Predict Batch'):
+    response=requests.post("https://siddhesh1981-CustomerChurnBackend.hf.space/Predict/Batch",files={'file':file2})
+    if response.status_code==200:
+      result=response.json()
+      st.subheader('Batch Prediction Result')
+      st.success(result)
+    else:
+      st.error(response.text)
+
+
+
+                           
+
+                           

requirements.txt

@@ -1,3 +1,3 @@
-altair
-pandas
-streamlit
+pandas==2.2.2
+requests==2.28.1
+streamlit==1.43.2