prism-api-backend / backend /test_ai_load.py
TanishkOrion's picture
Deploy isolated API Backend
e1404c1
Raw
History Blame Contribute Delete
861 Bytes
import sys, os
# Same logic as video.py
ROOT_DIR = os.path.abspath(os.path.join(os.path.dirname(__file__), "../"))
sys.path.append(ROOT_DIR)
print(f"ROOT_DIR: {ROOT_DIR}")
print(f"Looking for ai_models at: {os.path.join(ROOT_DIR, 'ai_models')}")
print(f"ai_models exists: {os.path.exists(os.path.join(ROOT_DIR, 'ai_models'))}")
print(f"assets exists: {os.path.exists(os.path.join(ROOT_DIR, 'ai_models/prism/assets'))}")
try:
from ai_models.prism.feature_engine import PrismFeatureExtractor
from ai_models.prism.router import PrismRouter
prism_extractor = PrismFeatureExtractor()
prism_router = PrismRouter(
artifacts_dir=os.path.join(ROOT_DIR, "ai_models/prism/assets")
)
print("SUCCESS: AI models loaded OK")
except Exception as e:
import traceback
print(f"ERROR: {type(e).__name__}: {e}")
traceback.print_exc()