JUGGHM commited on
Commit
430e224
1 Parent(s): 917351d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -54,7 +54,7 @@ def depth_normal(img, model_selection="vit-small"):
54
  if model_selection == "vit_small":
55
  model = model_small
56
  cfg = cfg_small
57
- elif model == "vit_large":
58
  model = model_large
59
  cfg = cfg_large
60
 
 
54
  if model_selection == "vit_small":
55
  model = model_small
56
  cfg = cfg_small
57
+ elif model_selection == "vit_large":
58
  model = model_large
59
  cfg = cfg_large
60