bpiyush commited on
Commit
97dde8d
·
1 Parent(s): d54408e
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -63,6 +63,10 @@ if __name__ == "__main__":
63
  paths = [join(repo_path, "data/videos", c + ".csv") for c in cdf.DIR_NAME]
64
  dfs = [load_category_df(p, a) for p, a in zip(paths, action_names)]
65
  df = pd.concat(dfs, ignore_index=True)
 
 
 
 
66
  st.session_state.df = df
67
  else:
68
  df = st.session_state.df
 
63
  paths = [join(repo_path, "data/videos", c + ".csv") for c in cdf.DIR_NAME]
64
  dfs = [load_category_df(p, a) for p, a in zip(paths, action_names)]
65
  df = pd.concat(dfs, ignore_index=True)
66
+
67
+ # df["annot_path"] = df[["video_id", "category"]].apply(lambda x: join(repo_path, f"data/annotations/{x[1]}/{x[0]}.fps1.csv"), axis=1)
68
+ # df = df[df["annot_path"].apply(exists)]
69
+
70
  st.session_state.df = df
71
  else:
72
  df = st.session_state.df