Update app.py

app.py

@@ -18,6 +18,8 @@ def predict_depth(model, image):
     return model(image)
 
 def make_video(video_path, outdir='./vis_video_depth',encoder='vitl'):
+    if encoder not in ["vitl","vitb","vits"]:
+        encoder = "vitl"
     # DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
     # model = DepthAnything.from_pretrained('LiheYoung/depth_anything_vitl14').to(DEVICE).eval()
     # Define path for temporary processed frames
@@ -149,7 +151,7 @@ with gr.Blocks(css=css) as demo:
 
     with gr.Row():
         input_video = gr.Video(label="Input Video")
-        model_type = gr.Dropdown(["vits", "vitl"], type="value", label='Model Type')
+        model_type = gr.Dropdown(["vits", "vitb", "vitl"], type="value", label='Model Type')
     submit = gr.Button("Submit")
     processed_video = gr.Video(label="Processed Video")