crislmfroes commited on
Commit
5643665
1 Parent(s): 61ff316

Change base image and dependencies

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. requirements.txt +3 -1
Dockerfile CHANGED
@@ -1,4 +1,4 @@
1
- FROM huggingface/transformers-cpu
2
 
3
  WORKDIR /code
4
 
 
1
+ FROM python:3.9
2
 
3
  WORKDIR /code
4
 
requirements.txt CHANGED
@@ -8,4 +8,6 @@ pyowm
8
  gradio_tools
9
  audio2numpy
10
  numpy
11
- scipy
 
 
 
8
  gradio_tools
9
  audio2numpy
10
  numpy
11
+ scipy
12
+ torch
13
+ torchvision