Make it work

views.py

@@ -8,7 +8,7 @@ from streamlit_plotly_events import plotly_events
 import utils
 import pandas as pd
 from scipy.spatial import distance
-
+from resources import get_gtr_embeddings
 dimensionality_reduction_model_name = "PCA"
 
 def diffs(embeddings: np.ndarray, corrector, encoder, tokenizer):
@@ -26,15 +26,22 @@ def diffs(embeddings: np.ndarray, corrector, encoder, tokenizer):
     with st.form(key="foo") as form:
         submit_button = st.form_submit_button("Synthesize")
 
-        sent1 = st.text_input("Sentence 1")
+        sent1 = st.text_input("Sentence 1", value="I am a king")
         st.latex("-")
-        sent2 = st.text_input("Sentence 2")
+        sent2 = st.text_input("Sentence 2", value="I am a man")
         st.latex("+")
-        sent3 = st.text_input("Sentence 3")
+        sent3 = st.text_input("Sentence 3", value="I am a woman")
         st.latex("=")
 
         if submit_button:
-            generated_sentence = "HI"
+            v1, v2, v3 = get_gtr_embeddings([sent1, sent2, sent3], encoder, tokenizer).to("cpu")
+            v4 = v1 - v2 + v3
+            generated_sentence, = vec2text.invert_embeddings(
+                embeddings=v4.unsqueeze(0).cuda(),
+                corrector=corrector,
+                num_steps=20,
+            )
+            generated_sentence = generated_sentence.strip()
         
         sent4 = st.text_input("Sentence 4", value=generated_sentence, disabled=True)