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())}
|