Update app.py

app.py

@@ -467,7 +467,7 @@ def encode_space(x):
     im_emb, _ = pipe.encode_image(
                 image, DEVICE, 1, output_hidden_state
             )
-    return im_emb
+    return im_emb.detach().to('cpu')
 
 # prep our calibration prompts
 for im in [