deveix commited on
Commit
18e733f
·
1 Parent(s): 5da4449

fix requirements

Browse files
Files changed (2) hide show
  1. app/main.py +3 -3
  2. requirements.txt +1 -1
app/main.py CHANGED
@@ -9,6 +9,8 @@ from dotenv import load_dotenv
9
  from fastapi.middleware.cors import CORSMiddleware
10
  from uuid import uuid4
11
  # import httpx
 
 
12
 
13
  import joblib
14
  import librosa
@@ -22,9 +24,7 @@ import opensmile
22
 
23
  import ffmpeg
24
  import noisereduce as nr
25
- from tensorflow.keras.models import load_model
26
- from tensorflow.keras.utils import to_categorical
27
- from tensorflow.keras.models import Sequential
28
 
29
  default_sample_rate=22050
30
 
 
9
  from fastapi.middleware.cors import CORSMiddleware
10
  from uuid import uuid4
11
  # import httpx
12
+ from tensorflow import keras
13
+ from tensorflow.keras.models import load_model
14
 
15
  import joblib
16
  import librosa
 
24
 
25
  import ffmpeg
26
  import noisereduce as nr
27
+
 
 
28
 
29
  default_sample_rate=22050
30
 
requirements.txt CHANGED
@@ -9,7 +9,7 @@ pypdf==4.0.2
9
  pymongo>=3.11
10
  tiktoken==0.6.0
11
  langchain-openai==0.0.8
12
- tensorflow
13
  python-dotenv
14
  upstash-redis
15
  librosa
 
9
  pymongo>=3.11
10
  tiktoken==0.6.0
11
  langchain-openai==0.0.8
12
+ tensorflow>=2.5,<2.6
13
  python-dotenv
14
  upstash-redis
15
  librosa