Update app.py

app.py

@@ -5,18 +5,9 @@ import modin.pandas as pd
 import numpy as np
 from diffusers import DiffusionPipeline 
 
-device = "cuda" if torch.cuda.is_available() else "cpu"
-
-if torch.cuda.is_available():
-    torch.cuda.max_memory_allocated(device=device)
-    torch.cuda.empty_cache()
-    pipe = DiffusionPipeline.from_pretrained("mann-e/Mann-E_Dreams", torch_dtype=torch.float16)
-    pipe.enable_xformers_memory_efficient_attention()
-    pipe = pipe.to(device)
-    torch.cuda.empty_cache()
-else: 
-    pipe = DiffusionPipeline.from_pretrained("mann-e/Mann-E_Dreams", use_safetensors=True)
-    pipe = pipe.to(device)
+pipe = DiffusionPipeline.from_pretrained("mann-e/Mann-E_Dreams", torch_dtype=torch.float16).to("cuda")
+pipe.enable_xformers_memory_efficient_attention()
+torch.cuda.empty_cache()
 
 @spaces.GPU
 def genie (prompt, negative_prompt, steps, seed):