kotlarska2 commited on
Commit
cd08094
·
verified ·
1 Parent(s): 28bc16c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -148,7 +148,7 @@ def display_images(folder_path, image_names):
148
  # Cast 6 -------------------------------------------------------------------------
149
 
150
  # Nastavenie parametrov pre funkciu process_input
151
- images_folder = "kotlarska2/Trains"
152
  device = "cuda" if torch.cuda.is_available() else "cpu"
153
  embeddings = get_clip_embeddings_batch(images_folder, 32, device)
154
 
 
148
  # Cast 6 -------------------------------------------------------------------------
149
 
150
  # Nastavenie parametrov pre funkciu process_input
151
+ images_folder = "Datasets/kotlarska2/Trains"
152
  device = "cuda" if torch.cuda.is_available() else "cpu"
153
  embeddings = get_clip_embeddings_batch(images_folder, 32, device)
154