update this

Dockerfile

@@ -6,22 +6,23 @@ WORKDIR /app
 RUN apt-get update && apt-get install -y \
     build-essential \
     curl \
+    software-properties-common \
     git \
     && rm -rf /var/lib/apt/lists/*
 
 # Copy files
-COPY requirements.txt .
-COPY app.py .
-COPY resnet18_cat.pth .
+COPY requirements.txt ./
+COPY resnet18_cat.pth ./
+COPY src/ ./src/
 
 # Install Python dependencies
-RUN pip install --no-cache-dir -r requirements.txt
+RUN pip3 install -r requirements.txt
 
 # Expose Streamlit default port
-EXPOSE 7860
+EXPOSE 8501
 
 # Healthcheck (optional)
-HEALTHCHECK CMD curl --fail http://localhost:7860/_stcore/health || exit 1
+HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
 
 # Run the Streamlit app using app.py
-CMD ["streamlit", "run", "app.py", "--server.port=7860", "--server.address=0.0.0.0"]
+ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]

README.md

@@ -3,10 +3,13 @@ title: Cat XAI App
 emoji: 🐱
 colorFrom: pink
 colorTo: indigo
-sdk: streamlit
-sdk_version: "1.31.1"
-app_file: app.py
+sdk: docker
+app_port: 8501
+tags:
+  - streamlit
 pinned: false
+short_description: Streamlit template space
+license: mit
 ---
 
 # Explainable AI for Image Classification using Twin System and Grad-CAM