Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 import torch
 from diffusers import FluxPipeline, FluxTransformer2DModel, FlowMatchEulerDiscreteScheduler
@@ -5,7 +6,10 @@ from huggingface_hub import hf_hub_download
 from PIL import Image
 import numpy as np
 import random
-import spaces
+
+
+# Only initialize GPU after spaces import
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 # Constants
 #BASE_MODEL = "black-forest-labs/FLUX.1-dev"