feat: AE-811 Add a class to create network volume using REST

[pandyamarut]

import asyncio
import base64
import io
from PIL import Image
from tetra_rp import remote, LiveServerless, GpuGroup

# Configuration for a GPU resource
sd_config = LiveServerless(
    gpus=[GpuGroup.AMPERE_80],
    name="example_image_gen_server",
)


@remote(
    resource_config=sd_config,
    dependencies=["diffusers", "transformers", "torch", "accelerate", "safetensors"],
    mount_volume=NetworkVolume(
        size=10,  # Size in GB
        dataCenterId="EU-RO-1",
        name="tetra-network-volume",
    ), # or You can pass it as: 
    
@remote(
    resource_config=sd_config,
    dependencies=["diffusers", "transformers", "torch", "accelerate", "safetensors"],
    mount_volume=NetworkVolume(
        id="network_volume_id",
    ),
)


)
def test_run():
    return "Hello from remote serverless function!"

async def main():
    test_run_result = await test_run()
    print(f"Remote function result: {test_run_result}")
    
if __name__ == "__main__":
    asyncio.run(main())

With this PR, you can mount the network volume to use with tetra function classes.