feat: remove hidden_state

app.py

@@ -38,8 +38,8 @@ def generate(query1, query2, source_code):
     # 特殊处理 Salesforce/codet5p-110m-embedding 模型
     inputs = tokenizer([query1, query2, source_code], padding=True, truncation=True, return_tensors="pt")
     with torch.no_grad():
-        outputs = model6(**inputs)
-        embeddings = outputs.last_hidden_state[:, 0, :].cpu().numpy()  # 使用 [CLS] token 的嵌入
+        embeddings = model6(**inputs)  # 这里直接返回嵌入向量
+        embeddings = embeddings.cpu().numpy()  # 转换为 NumPy 数组
 
     score1 = cos_sim(embeddings[0], embeddings[2])
     score2 = cos_sim(embeddings[1], embeddings[2])