init

app.py

@@ -33,6 +33,7 @@
 #       step_fn: src.diffusion.stateful_flow_matching.sampling.ode_step_fn
 import random
 import os
+import spaces
 import torch
 import argparse
 from omegaconf import OmegaConf
@@ -43,7 +44,7 @@ from src.diffusion.flow_matching.scheduling import LinearScheduler
 from PIL import Image
 import gradio as gr
 import tempfile
-import spaces
+
 
 from huggingface_hub import snapshot_download