Update app.py

app.py

@@ -10,10 +10,10 @@ import os
 from PIL import Image
 import numpy as np
 
+device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
 def load_models(model_path="MeissonFlow/Meissonic", 
-            transformer_path="MeissonFlow/Muddit",
-            device="cuda"):
+            transformer_path="MeissonFlow/Muddit"):
     model = SymmetricTransformer2DModel.from_pretrained(
         transformer_path,
         subfolder="1024/transformer",