daniel
commited on
Commit
Β·
887532d
1
Parent(s):
7034c1e
Fix numpy version and load_models
Browse files- api/dependencies.py +1 -2
- requirements.txt +1 -0
api/dependencies.py
CHANGED
|
@@ -20,8 +20,7 @@ async def initialize_models():
|
|
| 20 |
"""Initialize models on startup"""
|
| 21 |
global model_service
|
| 22 |
print("π Loading CodeBERT models...")
|
| 23 |
-
model_service = ModelService()
|
| 24 |
-
await model_service.load_models()
|
| 25 |
print("β
Models loaded successfully!")
|
| 26 |
|
| 27 |
|
|
|
|
| 20 |
"""Initialize models on startup"""
|
| 21 |
global model_service
|
| 22 |
print("π Loading CodeBERT models...")
|
| 23 |
+
model_service = ModelService() # Models load in __init__
|
|
|
|
| 24 |
print("β
Models loaded successfully!")
|
| 25 |
|
| 26 |
|
requirements.txt
CHANGED
|
@@ -5,3 +5,4 @@ torch==2.1.1
|
|
| 5 |
pydantic==2.5.0
|
| 6 |
python-multipart==0.0.6
|
| 7 |
huggingface-hub>=0.19.0
|
|
|
|
|
|
| 5 |
pydantic==2.5.0
|
| 6 |
python-multipart==0.0.6
|
| 7 |
huggingface-hub>=0.19.0
|
| 8 |
+
numpy<2
|