Update app.py

app.py

@@ -45,14 +45,14 @@ if text1 and text2:
     fine_tuned_out = cosine_similarity(
         [mean_pooled[0]],
         mean_pooled[1:]
-    )[0]
+    )[0][0]
     
     outputs_base = model_base(**tokens)
     mean_pooled_base = pool_embeddings(outputs_base,tokens).detach().numpy()
     base_out = cosine_similarity(
         [mean_pooled_base[0]],
         mean_pooled_base[1:]
-    )[0]
+    )[0][0]
     
     st.write(f">>>Similarity for fine-tuned {fine_tuned_out}")
     st.write(f">>>Similarity for base {base_out}")