Spaces:

traversaal-internal
/

nomic-embedding

Sleeping

AreejMehboob commited on 16 days ago

Commit

7603e25

verified ·

1 Parent(s): d47acd6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -91,6 +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().cpu().numpy()
 # :white_check_mark: Gradio interface function

     with torch.no_grad():  # Disable gradient calculation for inference
         outputs = text_model(**inputs)
     embeddings = outputs.last_hidden_state.mean(dim=1)
+    print(embeddings[0].detach().cpu().numpy())
     return embeddings[0].detach().cpu().numpy()
 # :white_check_mark: Gradio interface function