Update app.py
Browse files
app.py
CHANGED
@@ -17,7 +17,9 @@ Explore Vision Transformers, their architectures, and tokenization mechanisms.
|
|
17 |
# Model Selection
|
18 |
model_name = st.selectbox(
|
19 |
"Choose a Vision Transformer Model:",
|
20 |
-
["beit-base-patch16
|
|
|
|
|
21 |
)
|
22 |
|
23 |
# Load Tokenizer & Model
|
|
|
17 |
# Model Selection
|
18 |
model_name = st.selectbox(
|
19 |
"Choose a Vision Transformer Model:",
|
20 |
+
["microsoft/beit-base-patch16-224",
|
21 |
+
"microsoft/swin-base-patch4-window7",
|
22 |
+
"google/vit-base-patch16"]
|
23 |
)
|
24 |
|
25 |
# Load Tokenizer & Model
|