Hugging Face's logo

Spaces:
AlyxTeam
/
code_embedding_test
Running

App Files Community
zhangxiyi.amos commited on
Commit
dab847f
·
1 Parent(s): 94a6df5

feat: 获取模型

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -14,9 +14,8 @@ model4 = SentenceTransformer("aspire/acge_text_embedding")
14
  model5 = SentenceTransformer("intfloat/multilingual-e5-large")
15
 
16
  # 对于 Salesforce/codet5p-110m-embedding 模型，我们需要特殊处理
17
- config = AutoConfig.from_pretrained("Salesforce/codet5p-110m-embedding", trust_remote_code=True)
18
  tokenizer = AutoTokenizer.from_pretrained("Salesforce/codet5p-110m-embedding", trust_remote_code=True)
19
- model6 = AutoModel.from_pretrained("Salesforce/codet5p-110m-embedding", config=config, trust_remote_code=True)
20
 
21
  @spaces.GPU
22
  def generate(query1, query2, source_code):
 
14
  model5 = SentenceTransformer("intfloat/multilingual-e5-large")
15
 
16
  # 对于 Salesforce/codet5p-110m-embedding 模型，我们需要特殊处理
 
17
  tokenizer = AutoTokenizer.from_pretrained("Salesforce/codet5p-110m-embedding", trust_remote_code=True)
18
+ model6 = AutoModel.from_pretrained("Salesforce/codet5p-110m-embedding", trust_remote_code=True)
19
 
20
  @spaces.GPU
21
  def generate(query1, query2, source_code):