Update app.py
Browse filesusing only bert without head
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)}")
|