Changed classification model

Dockerfile

@@ -3,17 +3,20 @@ FROM python:3.10-slim
 
 # Install system dependencies
 RUN apt-get update && \
-    apt-get install -y wget ffmpeg unzip curl gcc portaudio19-dev && \
+    apt-get install -y wget ffmpeg unzip curl gcc  && \
     rm -rf /var/lib/apt/lists/*
 
 # Set working directory
 WORKDIR /app
+# portaudio19-dev
 
 # Copy your code into the container
 COPY . .
 
 # Install Python dependencies
 RUN pip install --upgrade pip && pip install -r requirements.txt
+RUN pip install --no-cache-dir torch==2.2.0+cpu torchvision==0.17.0+cpu torchaudio==2.2.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
+