Update app.py

app.py

@@ -91,7 +91,7 @@ def get_text_embeddings(text):
     with torch.no_grad():  # Disable gradient calculation for inference
         outputs = text_model(**inputs)
     embeddings = outputs.last_hidden_state.mean(dim=1)
-    return embeddings[0].detach().numpy()
+    return embeddings[0].detach().cpu().numpy()
 
 # :white_check_mark: Gradio interface function
 def embed_text_interface(text):