Spaces:
Paused
Paused
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
|