Update app.py
Browse files
app.py
CHANGED
@@ -81,7 +81,7 @@ def preprocess_images(examples):
|
|
81 |
return {"pixel_values": [process_image(img) for img in examples["image"]]}
|
82 |
|
83 |
# Load dataset with caching
|
84 |
-
dataset_path = "
|
85 |
dataset_cache_file = os.path.join(cache_dir, "montevideo_dataset.pkl")
|
86 |
|
87 |
print(f"Dataset path: {dataset_path}")
|
|
|
81 |
return {"pixel_values": [process_image(img) for img in examples["image"]]}
|
82 |
|
83 |
# Load dataset with caching
|
84 |
+
dataset_path = "uruguayai/montevideo"
|
85 |
dataset_cache_file = os.path.join(cache_dir, "montevideo_dataset.pkl")
|
86 |
|
87 |
print(f"Dataset path: {dataset_path}")
|