Spaces:

VashuTheGreat2
/

ML-Learner

Sleeping

ML-Learner / python_backend /src /Predictors /constants /__init__.py

Upload folder using huggingface_hub

c01955c verified 7 months ago

467 Bytes

import os

import torch
# ----------- JOb Similar -------------------------
MLFLOW_TRACKING_URI=os.getenv('ML_FLOW_TRACKING_URI')
MLFLOW_TRACKING_USERNAME=os.getenv("MLFLOW_TRACKING_USERNAME")
MLFLOW_TRACKING_PASSWORD=os.getenv("MLFLOW_TRACKING_PASSWORD")
MLFLOW_MODEL_URI= "models:/JobSimilarity/2"
MLFLOW_DOWNLOADED_MODEL_PATH:str=os.path.join("artifact","JobSimilarityModel","JobSimilarity.pth")
DEVICE=torch.device("cuda" if torch.cuda.is_available() else "cpu")

	import os

	import torch
	# ----------- JOb Similar -------------------------
	MLFLOW_TRACKING_URI=os.getenv('ML_FLOW_TRACKING_URI')
	MLFLOW_TRACKING_USERNAME=os.getenv("MLFLOW_TRACKING_USERNAME")
	MLFLOW_TRACKING_PASSWORD=os.getenv("MLFLOW_TRACKING_PASSWORD")
	MLFLOW_MODEL_URI= "models:/JobSimilarity/2"
	MLFLOW_DOWNLOADED_MODEL_PATH:str=os.path.join("artifact","JobSimilarityModel","JobSimilarity.pth")
	DEVICE=torch.device("cuda" if torch.cuda.is_available() else "cpu")