Update app.py

app.py

@@ -14,7 +14,7 @@ import spaces
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 dtype = torch.bfloat16
 
-@spaces.GPU
+
 def load_models(model_path="MeissonFlow/Meissonic", 
             transformer_path="MeissonFlow/Muddit"):
     model = SymmetricTransformer2DModel.from_pretrained(
@@ -34,11 +34,11 @@ def load_models(model_path="MeissonFlow/Meissonic",
         transformer=model,
         scheduler=scheduler,
     )
-    pipe.to(device)
     return pipe
 
 # Load models (global variable to avoid reloading)
 pipe = load_models()
+pipe.to(device)
 
 # Common transform
 def get_transform(resolution):