Update app.py

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 st.set_page_config(layout="wide")
 
-DEBUG = True
+DEBUG = False
 if DEBUG:
     # Use some dummy data
     tokens = [
@@ -49,7 +49,6 @@ else:
     token_similarities = F.normalize(embeddings[0], dim=1) @ F.normalize(embeddings[1], dim=1).T
 
     sentence_similarity = F.normalize(torch.mean(embeddings[0], dim=0), dim=-1) @ F.normalize(torch.mean(embeddings[1], dim=0), dim=-1)
-    sentence_similarity = max(0, round(sentence_similarity.item(), 2))
 
     #print("="*60)
     #print("Mapping sentence1 to sentence2...")
@@ -67,6 +66,11 @@ else:
         i = torch.argmax(token_probabilities_21[j])
         #print(tokens[1][j].ljust(15), tokens[0][i].ljust(15), round(token_probabilities_21[j][i].item(), 2))
 
+    # Convert to naive python objects
+    sentence_similarity = max(0, round(sentence_similarity.item(), 2))
+    token_probabilities_12 = token_probabilities_12.numpy().tolist()
+    token_probabilities_21 = token_probabilities_21.numpy().tolist()
+
 # Simplify the tokens for display
 tokens = [[token[3:].replace("\u2581", " ") for token in sentence] for sentence in tokens]