MusIre commited on
Commit
e137b2e
·
verified ·
1 Parent(s): f575a77

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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())}