Update app.py

app.py

@@ -124,16 +124,18 @@ pipe.to(device=DEVICE)
 #assert len(output.frames[0]) == 16
 #leave_im_emb.detach().to('cpu')
 
-@spaces.GPU(duration=10)
+@spaces.GPU(duration=20)
 def generate_gpu(in_im_embs):
+    print('start gen')
     in_im_embs = in_im_embs.to('cuda').unsqueeze(0).unsqueeze(0)
     #im_embs = torch.cat((torch.zeros(1, 1280, device=DEVICE, dtype=dtype), in_im_embs), 0)
 
     output = pipe(prompt='a scene', guidance_scale=0, added_cond_kwargs={}, ip_adapter_image_embeds=[in_im_embs], num_inference_steps=STEPS)
-
+    print('image is made')
     im_emb, _ = pipe.encode_image(
-                output.frames[0][len(output.frames[0])//2], DEVICE, 1, output_hidden_state
+                output.frames[0][len(output.frames[0])//2], 'cuda', 1, output_hidden_state
             )
+    print('im_emb is made')
     im_emb = im_emb.detach().to('cpu').to(torch.float32)
     return output, im_emb