MusIre commited on
Commit
18c0502
·
verified ·
1 Parent(s): 2312979

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -50,7 +50,7 @@ data_transforms = transforms.Compose([
50
  ])
51
 
52
  # Load dataset
53
- csv_file = "cleaned_classes.csv"
54
  dataset = ArtDataset(csv_file=csv_file, transform=data_transforms)
55
 
56
  # Define model
 
50
  ])
51
 
52
  # Load dataset
53
+ csv_file = "classes.csv"
54
  dataset = ArtDataset(csv_file=csv_file, transform=data_transforms)
55
 
56
  # Define model