Spaces:
Sleeping
Sleeping
Fix PYTHONPATH in Dockerfile and correct EMBEDDING_MODEL_PATH in config.py
Browse files- Dockerfile +1 -1
- 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
|
| 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 = "
|
| 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"
|