Metin commited on
Commit
9e1812c
·
1 Parent(s): e4955ba

Fix PYTHONPATH in Dockerfile and correct EMBEDDING_MODEL_PATH in config.py

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. src/config.py +1 -1
Dockerfile CHANGED
@@ -19,6 +19,6 @@ EXPOSE 8501
19
 
20
  HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
21
 
22
- ENV PYTHONPATH="/app/src:${PYTHONPATH}"
23
 
24
  ENTRYPOINT ["streamlit", "run", "src/demo.py", "--server.port=8501", "--server.address=0.0.0.0"]
 
19
 
20
  HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
21
 
22
+ ENV PYTHONPATH="/app/src"
23
 
24
  ENTRYPOINT ["streamlit", "run", "src/demo.py", "--server.port=8501", "--server.address=0.0.0.0"]
src/config.py CHANGED
@@ -2,7 +2,7 @@ from pydantic_settings import BaseSettings
2
 
3
 
4
  class Config(BaseSettings):
5
- EMBEDDING_MODEL_PATH: str = "/models/embedding/gte-multilingual-base"
6
  TRAINING_DATA_PATH: str = "input/train_data_with_embeddings.parquet"
7
 
8
  GNN_MODEL_PATH: str = "models/undirected_gnn/gnn_classifier_model.pth"
 
2
 
3
 
4
  class Config(BaseSettings):
5
+ EMBEDDING_MODEL_PATH: str = "models/embedding/gte-multilingual-base"
6
  TRAINING_DATA_PATH: str = "input/train_data_with_embeddings.parquet"
7
 
8
  GNN_MODEL_PATH: str = "models/undirected_gnn/gnn_classifier_model.pth"