basically delirious lol

app.py

@@ -139,6 +139,7 @@ def generate_gpu(in_im_embs):
     im_emb = im_emb.detach().to('cpu').to(torch.float32)
     return output, im_emb
 
+@spaces.GPU()
 def generate(in_im_embs):
     output, im_emb = generate_gpu(in_im_embs)
     nsfw = maybe_nsfw(output.frames[0][len(output.frames[0])//2])