Update app.py

app.py

@@ -587,7 +587,7 @@ def encode_space(x, im):
         gemb = pali.vision_tower(im.to(dtype).to('cuda')).last_hidden_state
         
         print('pali_enced')
-        return im_emb.to(torch.float32), gemb.to(torch.float32)
+        return im_emb.to(torch.float32).to('cpu'), gemb.to(torch.float32).to('cpu')
 
 demo.launch(share=True,)