Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
|