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