Update app.py

app.py

@@ -51,10 +51,10 @@ model_large.to(device)
 model_small.to(device)
 
 def depth_normal(img, model_selection="vit-small"):
- if model_selection == "vit_small":
+ if model_selection == "vit-small":
  model = model_small
  cfg = cfg_small
- elif model_selection == "vit_large":
+ elif model_selection == "vit-large":
  model = model_large
  cfg = cfg_large