Update app.py

app.py

@@ -1,8 +1,7 @@
 import gradio as gr
 import torch
-import spaces
 from diffusers import DiffusionPipeline
-from accelerate import load_checkpoint_and_dispatch
+from accelerate import init_empty_weights, load_checkpoint_and_dispatch, infer_auto_device_map
 
 # Configuración para usar bfloat16 y CUDA si está disponible
 dtype = torch.bfloat16
@@ -14,6 +13,9 @@ pipe = None
 def load_model():
     global pipe
     if pipe is None:
+        # Inicializar ZeroGPU antes de cargar el modelo
+        init_empty_weights()
+
         pipe = DiffusionPipeline.from_pretrained(
             "black-forest-labs/FLUX.1-schnell", 
             torch_dtype=dtype