Update app.py

app.py

@@ -74,7 +74,6 @@ def process_entity(batch, model, device):
 
     # Convert tensor to list of token IDs, and decode them into a readable sentence
     pseudo_sentence_decoded = bert_tokenizer.decode(pseudo_sentence[0].tolist(), skip_special_tokens=False)
-    st.write("Pseudo Sentence: ", pseudo_sentence_decoded)
 
     with torch.no_grad():
         outputs = spaBERT_model(#input_ids=input_ids,
@@ -124,6 +123,7 @@ def get_spaBert_embedding(entity):
             st.write("Got Bert embedding for: ", entity)
         return get_bert_embedding(entity)                            #Fallback in-case SpaBERT could not resolve entity to retrieve embedding. Rare-cases only.
     else:
+        st.write("Pseudo Sentnece:",pseudo_sentences[entity_index])
         if(dev_mode == True):
             st.write("Got SpaBert embedding for: ", entity)
         return spaBERT_embeddings[entity_index]