main

gradio_app.py

@@ -11,6 +11,7 @@ from safetensors.torch import load_file
 from networks import lora_flux
 from library import flux_utils, flux_train_utils_recraft as flux_train_utils, strategy_flux
 import logging
+import spaces
 
 # Set up logger
 logger = logging.getLogger(__name__)