Jonas Leeb commited on
Commit
c9fbca7
·
1 Parent(s): 7285400

requirements update

Browse files
Files changed (2) hide show
  1. app.py +0 -2
  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
- sklearn
8
  transformers
 
4
  datasets
5
  torch
6
  gensim
7
+ scikit-learn
8
  transformers