Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -112,7 +112,7 @@ class Echo(torchvision.datasets.VisionDataset):
|
|
| 112 |
else:
|
| 113 |
# Load video-level labels
|
| 114 |
with open(f"{self.root}/FileList.csv") as f:
|
| 115 |
-
data =
|
| 116 |
data["Split"].map(lambda x: x.upper())
|
| 117 |
|
| 118 |
if self.split != "ALL":
|
|
|
|
| 112 |
else:
|
| 113 |
# Load video-level labels
|
| 114 |
with open(f"{self.root}/FileList.csv") as f:
|
| 115 |
+
data = pd.read_csv(f)
|
| 116 |
data["Split"].map(lambda x: x.upper())
|
| 117 |
|
| 118 |
if self.split != "ALL":
|