File size: 552 Bytes
4bde888
 
80dcbaf
4bde888
80dcbaf
5d21a6b
4bde888
 
 
80dcbaf
4bde888
 
 
 
80dcbaf
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
import subprocess
import sys
import importlib

# List of dependencies to ensure are installed
deps = ["transformers", "torch", "fastapi", "uvicorn", "huggingface_hub", "accelerate"]

for dep in deps:
    try:
        importlib.import_module(dep)
    except ImportError:
        print(f"Installing missing dependency: {dep}")
        subprocess.check_call([sys.executable, "-m", "pip", "install", dep])

# Now delay-import your app
print("Dependencies installed. Loading server...")
import main  # <-- this is now safe to import AFTER accelerate exists