Spaces:
Running on Zero
Running on Zero
| """Optional ZeroGPU adapter; local runs do not import or require spaces.""" | |
| import spaces | |
| _pipeline = None | |
| def initialize(pipeline): | |
| global _pipeline | |
| # Load and place models during CUDA emulation, before reserving a shared GPU. | |
| pipeline.initialize() | |
| _pipeline = pipeline | |
| def startup(): | |
| # The pinned spaces SDK normally invokes this through Blocks.launch(). | |
| # Our mounted FastAPI application must report startup explicitly instead. | |
| from spaces.zero import startup as report_startup | |
| report_startup() | |
| def predict(images, camera, options, prepared=None): | |
| if _pipeline is None: | |
| raise RuntimeError('ZeroGPU models have not been initialized.') | |
| return _pipeline.predict(images, camera, options, prepared=prepared) | |