lyangas commited on
Commit
e980b69
·
verified ·
1 Parent(s): 1d4d02a

Update app.py

Browse files

using only bert without head

Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -15,6 +15,7 @@ print('INFO: loading models')
15
  try:
16
  with open('embedder/embedder.pkl', 'rb') as f:
17
  embedder = pickle.load(f)
 
18
  print('INFO: embedder loaded')
19
  except Exception as e:
20
  print(f"ERROR: loading embedder failed with: {str(e)}")
 
15
  try:
16
  with open('embedder/embedder.pkl', 'rb') as f:
17
  embedder = pickle.load(f)
18
+ embedder.embedder = embedder.embedder.bert
19
  print('INFO: embedder loaded')
20
  except Exception as e:
21
  print(f"ERROR: loading embedder failed with: {str(e)}")