Update app.py

app.py

@@ -22,8 +22,12 @@ if sentence1 and sentence2:
     with torch.no_grad():
         embeddings = model(input_ids['input_ids'], attention_mask=input_ids['attention_mask'])[0]
     
-    # calculate the cosine similarity between the embeddings
-    similarity = torch.nn.functional.cosine_similarity(embeddings[0], embeddings[1]).item()
-    
-    # display the predicted similarity to the user
-    st.write("Similarity score between the sentences:", similarity)
+    # check if both sentences have embeddings
+    if embeddings.shape[0] == 2:
+        # calculate the cosine similarity between the embeddings
+        similarity = torch.nn.functional.cosine_similarity(embeddings[0], embeddings[1]).item()
+        
+        # display the predicted similarity to the user
+        st.write("Similarity score between the sentences:", similarity)
+    else:
+        st.write("Unable to calculate similarity.")