Update app.py

app.py

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