CodeGovindz commited on
Commit
d8c5526
·
1 Parent(s): 6f58ce3

Fix: Add tensorflow and tf_keras for DeepFace compatibility

Browse files
Files changed (2) hide show
  1. Dockerfile +2 -1
  2. requirements.txt +2 -0
Dockerfile CHANGED
@@ -33,8 +33,9 @@ EXPOSE 7860
33
 
34
  # Set environment variables
35
  ENV PYTHONUNBUFFERED=1
36
- ENV TRANSFORMERS_CACHE=/app/.cache
37
  ENV HF_HOME=/app/.cache
 
 
38
 
39
  # Run the application
40
  CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
 
33
 
34
  # Set environment variables
35
  ENV PYTHONUNBUFFERED=1
 
36
  ENV HF_HOME=/app/.cache
37
+ ENV TF_USE_LEGACY_KERAS=0
38
+ ENV TF_ENABLE_ONEDNN_OPTS=0
39
 
40
  # Run the application
41
  CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
requirements.txt CHANGED
@@ -10,6 +10,8 @@ transformers==4.44.0
10
  librosa==0.10.2
11
  pillow==10.4.0
12
  deepface==0.0.92
 
 
13
  soundfile==0.12.1
14
  audioread==3.0.1
15
  speechbrain==1.0.0
 
10
  librosa==0.10.2
11
  pillow==10.4.0
12
  deepface==0.0.92
13
+ tensorflow==2.15.0
14
+ tf_keras==2.15.0
15
  soundfile==0.12.1
16
  audioread==3.0.1
17
  speechbrain==1.0.0