Update app.py

app.py

@@ -7,15 +7,10 @@ import spaces
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-if torch.cuda.is_available():
-    torch.cuda.max_memory_allocated(device=device)
-    pipe = AuraFlowPipeline.from_pretrained("AuraDiffusion/auradiffusion-v0.1a0",
-                                             torch_dtype=torch.float16)
-    pipe.enable_xformers_memory_efficient_attention()
-    pipe = pipe.to(device)
-else: 
-    pipe = DiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo", use_safetensors=True)
-    pipe = pipe.to(device)
+pipeline = AuraFlowPipeline(
+	"AuraDiffusion/auradiffusion-v0.1a0", 
+	torch_dtype=torch.float16
+).to("cuda")
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024