Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -21,7 +21,7 @@ else:
|
|
| 21 |
# Dataset class
|
| 22 |
class ArtDataset(Dataset):
|
| 23 |
def __init__(self, csv_file, transform=None):
|
| 24 |
-
self.annotations = pd.read_csv(csv_file, delimiter="
|
| 25 |
self.transform = transform
|
| 26 |
self.label_map_style = {style: idx for idx, style in enumerate(self.annotations['genre'].unique())}
|
| 27 |
self.label_map_artist = {artist: idx for idx, artist in enumerate(self.annotations['artist'].unique())}
|
|
|
|
| 21 |
# Dataset class
|
| 22 |
class ArtDataset(Dataset):
|
| 23 |
def __init__(self, csv_file, transform=None):
|
| 24 |
+
self.annotations = pd.read_csv(csv_file, delimiter=";")
|
| 25 |
self.transform = transform
|
| 26 |
self.label_map_style = {style: idx for idx, style in enumerate(self.annotations['genre'].unique())}
|
| 27 |
self.label_map_artist = {artist: idx for idx, artist in enumerate(self.annotations['artist'].unique())}
|