AndrzejCodahead

@Hasnain1997-ai, we haven't converted the models to TensorRT. For segmentation model you can try follow by this [notebook](https://colab.research.google.com/drive/1ZFdkdIAjD0ldhJ9TEhzTL1bndLyqm2Rd?usp=sharing&pli=1#scrollTo=VCIbfJ45XUbY), for the classification model, this is a slightly modified resnet18 you can...

Hi @Hasnain1997-ai , the pytorch file of segmentation model you could download under this [link](https://storage.googleapis.com/fishial-ml-resources/model_21_08_2023.pth) , but if you loocking for already converted torchscript file you may download from [here](https://storage.googleapis.com/fishial-ml-resources/segmentation_21_08_2023.ts).