borutokarma123 commited on
Commit
3272087
·
verified ·
1 Parent(s): 0cdb8e0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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", "swin-base-patch4-window7", "vit-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