Spaces:
Runtime error
Runtime error
| from huggingface_hub import from_pretrained_keras | |
| class CganCols: | |
| def __init__(self): | |
| self.model_0 = from_pretrained_keras('SupawitMarayat/pimthaigans-cgan-0010') | |
| self.model_1 = from_pretrained_keras('SupawitMarayat/pimthaigans-cgan-1121') | |
| self.model_2 = from_pretrained_keras('SupawitMarayat/pimthaigans-cgan-2232') | |
| self.model_3 = from_pretrained_keras('SupawitMarayat/pimthaigans-cgan-3343') | |
| self.model_4 = from_pretrained_keras('SupawitMarayat/pimthaigans-cgan-4454') | |
| self.model_5 = from_pretrained_keras('SupawitMarayat/pimthaigans-cgan-5565') | |
| self.model_6 = from_pretrained_keras('SupawitMarayat/pimthaigans-cgan-6676') | |
| self.model_7 = from_pretrained_keras('SupawitMarayat/pimthaigans-cgan-7787') | |
| self.model_cols = [self.model_0, self.model_1, self.model_2, self.model_3, self.model_4, self.model_5, | |
| self.model_6, self.model_7] | |
| def get_model(index: int): | |
| """ | |
| Returns the model based on the given index. | |
| Index ranges and corresponding model values: | |
| 0-10 -> 0 | |
| 11-21 -> 1 | |
| 22-32 -> 2 | |
| 33-43 -> 3 | |
| 44-54 -> 4 | |
| 55-65 -> 5 | |
| 66-76 -> 6 | |
| 77-87 -> 7 | |
| :param index: The index value. | |
| :return: The corresponding model value. | |
| """ | |
| if 0 <= index <= 10: | |
| return 0 | |
| elif 11 <= index <= 21: | |
| return 1 | |
| elif 22 <= index <= 32: | |
| return 2 | |
| elif 33 <= index <= 43: | |
| return 3 | |
| elif 44 <= index <= 54: | |
| return 4 | |
| elif 55 <= index <= 65: | |
| return 5 | |
| elif 66 <= index <= 76: | |
| return 6 | |
| elif 77 <= index <= 87: | |
| return 7 | |