mlabonne commited on
Commit
c6f6aa7
1 Parent(s): 403650d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -4
app.py CHANGED
@@ -112,13 +112,15 @@ def main():
112
  df = pd.DataFrame(columns=full_df.columns)
113
 
114
  # Toggles
115
- col1, col2, col3 = st.columns(3)
116
  with col1:
117
- show_phi = st.checkbox("Phi (2.8B)", value=True)
118
  with col2:
119
- show_mistral = st.checkbox("Mistral (7B)", value=True)
120
  with col3:
121
- show_other = st.checkbox("Other", value=True)
 
 
122
 
123
  # Create a DataFrame based on selected filters
124
  dfs_to_concat = []
@@ -127,6 +129,8 @@ def main():
127
  dfs_to_concat.append(full_df[full_df['Tags'].str.lower().str.contains('phi,|phi-msft,')])
128
  if show_mistral:
129
  dfs_to_concat.append(full_df[full_df['Tags'].str.lower().str.contains('mistral,')])
 
 
130
  if show_other:
131
  other_df = full_df[~full_df['Tags'].str.lower().str.contains('phi,|phi-msft,|mistral,')]
132
  dfs_to_concat.append(other_df)
 
112
  df = pd.DataFrame(columns=full_df.columns)
113
 
114
  # Toggles
115
+ col1, col2, col3, col4 = st.columns(4)
116
  with col1:
117
+ show_phi = st.checkbox("Phi (2.8B)", value=False)
118
  with col2:
119
+ show_mistral = st.checkbox("Mistral (7B)", value=False)
120
  with col3:
121
+ show_llama = st.checkbox("Llama 3 (8B)", value=True)
122
+ with col4:
123
+ show_other = st.checkbox("Other", value=False)
124
 
125
  # Create a DataFrame based on selected filters
126
  dfs_to_concat = []
 
129
  dfs_to_concat.append(full_df[full_df['Tags'].str.lower().str.contains('phi,|phi-msft,')])
130
  if show_mistral:
131
  dfs_to_concat.append(full_df[full_df['Tags'].str.lower().str.contains('mistral,')])
132
+ if show_llama:
133
+ dfs_to_concat.append(full_df[full_df['Tags'].str.lower().str.contains('llama,')])
134
  if show_other:
135
  other_df = full_df[~full_df['Tags'].str.lower().str.contains('phi,|phi-msft,|mistral,')]
136
  dfs_to_concat.append(other_df)