updated

Dockerfile

@@ -1,21 +1,20 @@
-# Use NVIDIA PyTorch image with CUDA 11.8 and cuDNN pre-installed
-FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
+# Use NVIDIA image with PyTorch, CUDA, cuDNN preinstalled
+FROM pytorch/pytorch:2.1.2-cuda11.8-cudnn8-runtime
 
-# Set up environment
+# Enable non-interactive install
 ENV DEBIAN_FRONTEND=noninteractive
 
 # Install dependencies
 RUN apt-get update && apt-get install -y \
-    python3-pip python3-dev ffmpeg git libsndfile1 \
+    git ffmpeg libsndfile1 python3-dev \
     && rm -rf /var/lib/apt/lists/*
 
-# Install Python packages
+# Copy and install Python dependencies
 COPY requirements.txt .
 RUN pip install --upgrade pip && pip install -r requirements.txt
 
-# Copy app
+# Copy app files
 COPY . /app
 WORKDIR /app
 
-# Expose
-CMD ["python3", "app.py"]
+CMD ["python3", "app.py"]

README.md

@@ -1,7 +1,6 @@
 ---
-title: Codingo Interview App
+title: Codingo
 sdk: docker
 app_file: app.py
-python_version: "3.10"
 hardware: gpu
 ---