JoJosmin commited on
Commit
b232392
1 Parent(s): 196c895

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -10,6 +10,7 @@ from transformers import pipeline
10
  import chromadb
11
  from sklearn.metrics.pairwise import euclidean_distances
12
  from sklearn.preprocessing import normalize
 
13
 
14
  # Load segmentation model
15
  segmenter = pipeline(model="mattmdjaga/segformer_b2_clothes")
 
10
  import chromadb
11
  from sklearn.metrics.pairwise import euclidean_distances
12
  from sklearn.preprocessing import normalize
13
+ from sklearn.metrics.pairwise import cosine_similarity
14
 
15
  # Load segmentation model
16
  segmenter = pipeline(model="mattmdjaga/segformer_b2_clothes")