Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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")
|