File size: 380 Bytes
2e237ce
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
from configuration import RESTART_IF_NO_GPU
from drivers.web.fastapi_app import create_app
from drivers.web.dependency_injection import setup_dependencies
import torch

if RESTART_IF_NO_GPU:
    if not torch.cuda.is_available():
        raise RuntimeError("No GPU available. Restarting the service is required.")

controllers = setup_dependencies()

app = create_app(controllers)