SONICS detect API (FastAPI wrapper, SpecTTTra-alpha-120s)

Dockerfile

@@ -11,9 +11,9 @@ ENV HOME=/home/user \
     HF_HOME=/home/user/.cache/huggingface
 WORKDIR /home/user/app
 
-RUN pip install --no-cache-dir --user torch --index-url https://download.pytorch.org/whl/cpu && \
-    pip install --no-cache-dir --user librosa soundfile numpy fastapi "uvicorn[standard]" python-multipart huggingface_hub timm pandas && \
-    pip install --no-cache-dir --user git+https://github.com/awsaf49/sonics.git
+RUN pip install --no-cache-dir --user torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu && \
+    pip install --no-cache-dir --user librosa soundfile numpy fastapi "uvicorn[standard]" python-multipart huggingface_hub timm pandas matplotlib tqdm scikit-learn fvcore && \
+    pip install --no-cache-dir --user --no-deps git+https://github.com/awsaf49/sonics.git
 
 # Bake model weights into the image so cold starts skip the download.
 RUN python -c "from sonics import HFAudioClassifier; HFAudioClassifier.from_pretrained('awsaf49/sonics-spectttra-alpha-120s')"