paolodegasperis commited on
Commit
9a553b0
1 Parent(s): 2ba659c

modifica script per percorsi aggiornati

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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]), similarities[i]) for i in top_indices] # Rimuove Path("img") /
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