Jonas Leeb
commited on
Commit
·
c9fbca7
1
Parent(s):
7285400
requirements update
Browse files- app.py +0 -2
- requirements.txt +1 -1
app.py
CHANGED
@@ -6,9 +6,7 @@ from sklearn.metrics.pairwise import cosine_similarity
|
|
6 |
from sklearn.preprocessing import normalize
|
7 |
from transformers import BertTokenizer, BertModel
|
8 |
import numpy as np
|
9 |
-
import json
|
10 |
from datasets import load_dataset
|
11 |
-
import os
|
12 |
from gensim.models import KeyedVectors
|
13 |
|
14 |
|
|
|
6 |
from sklearn.preprocessing import normalize
|
7 |
from transformers import BertTokenizer, BertModel
|
8 |
import numpy as np
|
|
|
9 |
from datasets import load_dataset
|
|
|
10 |
from gensim.models import KeyedVectors
|
11 |
|
12 |
|
requirements.txt
CHANGED
@@ -4,5 +4,5 @@ numpy
|
|
4 |
datasets
|
5 |
torch
|
6 |
gensim
|
7 |
-
|
8 |
transformers
|
|
|
4 |
datasets
|
5 |
torch
|
6 |
gensim
|
7 |
+
scikit-learn
|
8 |
transformers
|