Update app.py

app.py

@@ -104,6 +104,16 @@ def compute_embeddings(selected_task, input_text):
     clear_cuda_cache()
     return embeddings_list
 
+@spaces.GPU
+def decode_embedding(embedding_str):
+    try:
+        embedding = [float(num) for num in embedding_str.split(',')]
+        embedding_tensor = torch.tensor(embedding, dtype=torch.float16, device=device)
+        decoded_embedding = tokenizer.decode(embedding_tensor[0], skip_special_tokens=True)
+        return decoded_embedding.cpu().numpy().tolist()
+    except Exception as e:
+        return f"Error in decoding: {str(e)}"
+
 @spaces.GPU
 def compute_similarity(selected_task, sentence1, sentence2, extra_sentence1, extra_sentence2):
     try:
@@ -280,6 +290,17 @@ def app_interface():
                 outputs=output_display_connector
             )
 
+        with gr.Tab("Decode Embedding"):
+            embedding_input = gr.Textbox(label="Enter Embedding (comma-separated floats)")
+            decode_button = gr.Button("Decode")
+            decoded_output = gr.Textbox(label="Decoded Embedding")
+    
+            decode_button.click(
+                fn=decode_embedding,
+                inputs=embedding_input,
+                outputs=decoded_output
+            )
+
         with gr.Row():
             with gr.Column():
                 input_text_box