Spaces:
Runtime error
Runtime error
| # import the necessary packages | |
| from tensorflow import keras | |
| def loader(stem, trunk, attn): | |
| # load the models from disk | |
| conv_stem = keras.models.load_model( | |
| stem, | |
| compile=False | |
| ) | |
| conv_trunk = keras.models.load_model( | |
| trunk, | |
| compile=False | |
| ) | |
| conv_attn = keras.models.load_model( | |
| attn, | |
| compile=False | |
| ) | |
| return (conv_stem, conv_trunk, conv_attn) |