Update app.py

app.py

@@ -54,7 +54,7 @@ def depth_normal(img, model_selection="vit-small"):
     if model_selection == "vit_small":
         model = model_small
         cfg = cfg_small
-    elif model == "vit_large":
+    elif model_selection == "vit_large":
         model = model_large
         cfg = cfg_large