Update app.py

app.py

@@ -14,6 +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(