Update app.py

app.py

@@ -2,6 +2,7 @@ import os
 import gradio as gr
 import numpy as np
 from transformers import AutoTokenizer, AutoModel
+import time 
 # :white_check_mark: Setup environment
 os.makedirs(os.environ.get("HF_HOME", "./hf_cache"), exist_ok=True)
 hf_token = os.environ.get("HF_TOKEN")
@@ -32,7 +33,9 @@ def get_text_embeddings(text):
     return embeddings[0].detach().numpy()
 # :white_check_mark: Gradio interface function
 def embed_text_interface(text):
+    strt_time=time.time()
     embedding = get_text_embeddings(text)
+    print(f"Total time taken by nomic to embed: {time.time()-strt_time}")
     return str(embedding)
 # :white_check_mark: Gradio UI
 interface = gr.Interface(