Vivien commited on
Commit
da6fb88
1 Parent(s): fbe7708

Add the larger models

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -5,10 +5,7 @@ import pandas as pd, numpy as np
5
  from transformers import CLIPProcessor, CLIPModel
6
  from st_clickable_images import clickable_images
7
 
8
- MODEL_NAMES = [
9
- "base-patch32",
10
- "base-patch16",
11
- ] # , "large-patch14", "large-patch14-336"]
12
 
13
 
14
  @st.cache(allow_output_mutation=True)
@@ -177,7 +174,7 @@ def main():
177
  if "Comparison" in mode:
178
  c1, c2 = st.columns((1, 1))
179
  selection1 = c1.selectbox("", models_dict.keys(), index=0)
180
- selection2 = c2.selectbox("", models_dict.keys(), index=1)
181
  name1 = models_dict[selection1]
182
  name2 = models_dict[selection2]
183
  else:
 
5
  from transformers import CLIPProcessor, CLIPModel
6
  from st_clickable_images import clickable_images
7
 
8
+ MODEL_NAMES = ["base-patch32", "base-patch16", "large-patch14", "large-patch14-336"]
 
 
 
9
 
10
 
11
  @st.cache(allow_output_mutation=True)
 
174
  if "Comparison" in mode:
175
  c1, c2 = st.columns((1, 1))
176
  selection1 = c1.selectbox("", models_dict.keys(), index=0)
177
+ selection2 = c2.selectbox("", models_dict.keys(), index=3)
178
  name1 = models_dict[selection1]
179
  name2 = models_dict[selection2]
180
  else: