Update app.py

app.py

@@ -50,7 +50,7 @@ data_transforms = transforms.Compose([
 ])
 
 # Load dataset
-csv_file = "cleaned_classes.csv"
+csv_file = "classes.csv"
 dataset = ArtDataset(csv_file=csv_file, transform=data_transforms)
 
 # Define model