()
| 134 | |
| 135 | @st.cache_resource |
| 136 | def init_image_model(): |
| 137 | image_model_type = load_config('image', 'image_model_type') |
| 138 | image_model_config = load_config('image', 'image_model_config').get(image_model_type) |
| 139 | image_model = image_models[image_model_type](**image_model_config) |
| 140 | return image_model |
| 141 | |
| 142 | |
| 143 | def text_to_image(prompt, image_model): |