Update app.py

app.py

@@ -3,6 +3,7 @@ import torch
 import modin.pandas as pd
 import numpy as np
 from diffusers import DiffusionPipeline 
+import spaces 
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
@@ -16,7 +17,8 @@ if torch.cuda.is_available():
 else: 
     pipe = DiffusionPipeline.from_pretrained("mann-e/Mann-E_Dreams", use_safetensors=True)
     pipe = pipe.to(device)
-    
+
+@spaces.GPU
 def genie (prompt, negative_prompt, steps, seed):
     generator = np.random.seed(0) if seed == 0 else torch.manual_seed(seed)
     int_image = pipe(prompt=prompt, negative_prompt=negative_prompt, generator=generator, num_inference_steps=steps, guidance_scale=0.0).images[0]