daniel commited on
Commit
887532d
Β·
1 Parent(s): 7034c1e

Fix numpy version and load_models

Browse files
Files changed (2) hide show
  1. api/dependencies.py +1 -2
  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