Spaces:
Sleeping
Sleeping
paolodegasperis
commited on
Commit
•
9a553b0
1
Parent(s):
2ba659c
modifica script per percorsi aggiornati
Browse files
app.py
CHANGED
@@ -16,6 +16,8 @@ def load_openclip_model(device):
|
|
16 |
|
17 |
def load_embeddings(embedding_file):
|
18 |
df = pd.read_csv(embedding_file)
|
|
|
|
|
19 |
embeddings = df.iloc[:, 1:].values # Escludi la prima colonna (filename)
|
20 |
image_paths = df['filename'].tolist() # Salva i nomi dei file
|
21 |
return embeddings, image_paths
|
@@ -32,7 +34,7 @@ def query_images(text, model, preprocess, image_embeddings, image_paths, device,
|
|
32 |
top_indices = similarities.argsort()[-num_images:][::-1]
|
33 |
|
34 |
# Restituisci i percorsi delle immagini più simili e i loro punteggi
|
35 |
-
return [(Path(image_paths[i]
|
36 |
|
37 |
def predict(query_text, num_images):
|
38 |
# Ottieni il numero di immagini simili specificato e i loro punteggi
|
|
|
16 |
|
17 |
def load_embeddings(embedding_file):
|
18 |
df = pd.read_csv(embedding_file)
|
19 |
+
# Correggi i percorsi delle immagini per Linux
|
20 |
+
df['filename'] = df['filename'].str.replace("\\", "/")
|
21 |
embeddings = df.iloc[:, 1:].values # Escludi la prima colonna (filename)
|
22 |
image_paths = df['filename'].tolist() # Salva i nomi dei file
|
23 |
return embeddings, image_paths
|
|
|
34 |
top_indices = similarities.argsort()[-num_images:][::-1]
|
35 |
|
36 |
# Restituisci i percorsi delle immagini più simili e i loro punteggi
|
37 |
+
return [(Path("img") / image_paths[i], similarities[i]) for i in top_indices]
|
38 |
|
39 |
def predict(query_text, num_images):
|
40 |
# Ottieni il numero di immagini simili specificato e i loro punteggi
|