1TSnakers commited on
Commit
1c028c8
·
verified ·
1 Parent(s): 8ec4f90

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +18 -5
app.py CHANGED
@@ -40,15 +40,26 @@ if st.button("Make list"):
40
  for model in library:
41
  pull_count_str.append(model["pull_count_str"])
42
 
 
 
 
 
 
 
 
 
 
43
  if robot_mode:
44
  df = pd.DataFrame({
45
  "model": models,
46
- "pull_count": pull_count
 
47
  })
48
  else:
49
  df = pd.DataFrame({
50
  "model": models,
51
- "pull_count_str": pull_count_str
 
52
  })
53
  end_time = time.time()
54
  elapsed_time = end_time - start_time
@@ -60,8 +71,9 @@ if st.button("Make list"):
60
  df,
61
  hide_index=True,
62
  column_config={
63
- "name": "AI Model",
64
- "pull_count": "Pulls"
 
65
  }
66
  )
67
  else:
@@ -70,6 +82,7 @@ if st.button("Make list"):
70
  hide_index=True,
71
  column_config={
72
  "name": "AI Model",
73
- "pull_count_str": "Pulls"
 
74
  }
75
  )
 
40
  for model in library:
41
  pull_count_str.append(model["pull_count_str"])
42
 
43
+ # add date updated
44
+ date = []
45
+ for model in library:
46
+ date.append(model["last_updated_str"])
47
+
48
+ date_iso = []
49
+ for model in library:
50
+ date_iso.append(model["last_updated_iso"])
51
+
52
  if robot_mode:
53
  df = pd.DataFrame({
54
  "model": models,
55
+ "pull_count": pull_count,
56
+ "date_iso": date_iso
57
  })
58
  else:
59
  df = pd.DataFrame({
60
  "model": models,
61
+ "pull_count_str": pull_count_str,
62
+ "date": date
63
  })
64
  end_time = time.time()
65
  elapsed_time = end_time - start_time
 
71
  df,
72
  hide_index=True,
73
  column_config={
74
+ "model": "AI Model",
75
+ "pull_count": "Pulls",
76
+ "date_iso": "Last Updated"
77
  }
78
  )
79
  else:
 
82
  hide_index=True,
83
  column_config={
84
  "name": "AI Model",
85
+ "pull_count_str": "Pulls",
86
+ "date": "Last Updated"
87
  }
88
  )