OpenIPOS commited on
Commit
83b243a
·
verified ·
1 Parent(s): 0fe4fe7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import numpy as np
5
 
6
  # 1. 加载谷歌专利专用 BERT (0.3B 极其轻量)
7
  model_name = "anferico/bert-for-patents"
8
- tokenizer = AutoTokenizer.from_pretrained(model_name)
9
  model = AutoModel.from_pretrained(model_name)
10
 
11
  def get_embedding(text):
 
5
 
6
  # 1. 加载谷歌专利专用 BERT (0.3B 极其轻量)
7
  model_name = "anferico/bert-for-patents"
8
+ tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=False)
9
  model = AutoModel.from_pretrained(model_name)
10
 
11
  def get_embedding(text):