fix: 使用cpu获取codet5p嵌入表示

app.py

@@ -39,7 +39,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():
-        embeddings = model6(**inputs)[0]  # 直接使用模型输出的嵌入
+        outputs = model6(**inputs)
+        embeddings = outputs.last_hidden_state[:, 0, :].cpu().numpy()  # 使用 [CLS] token 的嵌入
 
     score1 = cos_sim(embeddings[0], embeddings[2])
     score2 = cos_sim(embeddings[1], embeddings[2])