Spaces:
Sleeping
Sleeping
from app import app | |
from flask import request, jsonify | |
def cluster(): | |
data = request.get_json() | |
embeddings = data['embeddings'] | |
from sklearn.cluster import DBSCAN | |
import numpy as np | |
try: | |
dbscan = DBSCAN(eps=0.4, min_samples=2, metric='cosine', n_jobs=-1) | |
embeddings_array = np.array(embeddings) | |
labels = dbscan.fit_predict(embeddings_array) | |
labels = labels.tolist() | |
return jsonify({'labels': labels}) | |
except Exception as e: | |
return jsonify({'error': str(e)}) | |