1from sentence_transformers import SentenceTransformer
2from sentence_transformers.util import cos_sim
3
4model1 = SentenceTransformer(
5 "perplexity-ai/pplx-embed-v1-0.6B",
6 trust_remote_code=True
7)
8model2 = SentenceTransformer("mykor/pplx-embed-v1-0.6b")
9
10text = """더워지는 바람에
11눈을 스쳐 뜨던 밤처럼
12옅어지는 그날의 작은 기억이
13잊혀져 매일 눈을 뜰 때면
14흐려져 오늘도 눈을 감으면
15또 사라져버릴 듯한 어제를 그려가
16
17떨어지는 그림자 사이에
18맴도는 향기가 조용히 너를 불어와
19선명했던 날들도 어느새
20다 지워버린 채
21차갑게 잊혀져만 가
22
23흐렸던 날들만 바람에 날아가거라
24베어 물은 듯 추억만 고이 남은 채
25지샌 하늘 위 피어진 구름처럼 사라지는
26마음은 후회도 잊어버린 채
27내 생에 피어라 가장 아픈 겨울아
28지난날처럼 길고 멀었던
29그리운 계절을 불러
30
31봄바람이 스치듯 떠난 밤
32내 안의 계절을 다 팔아 버린 밤
33마음에는 어떤 소리가 들려?
34아아
35
36깊어지는 실루엣 사이로
37눈 부신 바람이 또다시 너를 불어와
38선명했던 날들도 이제는
39다 잊어버린 채
40조용히 흩어져만 가
41
42괴로운 날들만 바람에 날아가거라
43베어 물은 듯 추억만 고이 남은 채
44지샌 새벽 끝 옅어진 달빛처럼 흐려지는
45기억은 슬픔도 잊어버린 채
46내 생에 지어라 가장 짙은 여름아
47지난날처럼 길고 멀었던
48그리운 계절을 불러
49
50하얗고 하얗던 내 계절아
51끝이 없고 그치지 않는 비에도
52밝아오니까
53그 시간이 두려워도
54난 괜찮아
55잿빛 사이 푸른 이 비가
56선명하게 모든 계절을
57다시 찾아갈 테니
58
59흐렸던 날들만 바람에 날아가거라
60베어 물은 듯 추억만 고이 남은 채
61푸른 바람과 스쳐간 계절마저 잊어가는
62시간은 어제도 잊어버린 채
63내게만 맑아라 슬피 우는 사랑아
64지난날처럼 길고 멀었던
65그리운 계절 아래로
66
67피어가 꽃잎과 푸른 하늘이
68베어 물은 듯 후회만 남아버린 채
69기쁜 마음도 슬픔도 이젠 되돌릴 수 없는
70날들로 저 멀리 사라져만 가
71내 생에 피었던 아름다운 하루가
72지난밤처럼 길고 어둡던
73그리운 계절을 불러"""
74
75embed1 = model1.encode(text)
76embed2 = model2.encode(text)
77print(cos_sim(embed1, embed2).item())