Spaces:

alaa-lab
/

Med-Real2Sim

Running

frances-dean commited on Mar 7

Commit

4d12343

•

1 Parent(s): b6fbf82

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -112,7 +112,7 @@ class Echo(torchvision.datasets.VisionDataset):
         else:
             # Load video-level labels
             with open(f"{self.root}/FileList.csv") as f:
-                data = pandas.read_csv(f)
             data["Split"].map(lambda x: x.upper())
             if self.split != "ALL":

         else:
             # Load video-level labels
             with open(f"{self.root}/FileList.csv") as f:
+                data = pd.read_csv(f)
             data["Split"].map(lambda x: x.upper())
             if self.split != "ALL":