JohanDL commited on
Commit
fdf9868
1 Parent(s): cb5ba84

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -17,7 +17,7 @@ from depth_anything.util.transform import Resize, NormalizeImage, PrepareForNet
17
  def predict_depth(model, image):
18
  return model(image)
19
 
20
- def make_video(video_path, outdir='./vis_video_depth',encoder='vits'):
21
  # DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
22
  # model = DepthAnything.from_pretrained('LiheYoung/depth_anything_vitl14').to(DEVICE).eval()
23
  # Define path for temporary processed frames
@@ -165,7 +165,7 @@ with gr.Blocks(css=css) as demo:
165
  example_files = os.listdir('assets/examples_video')
166
  example_files.sort()
167
  example_files = [os.path.join('assets/examples_video', filename) for filename in example_files]
168
- examples = gr.Examples(examples=example_files, inputs=[input_video, model_type], outputs=processed_video, fn=on_submit, cache_examples=True)
169
 
170
 
171
  if __name__ == '__main__':
 
17
  def predict_depth(model, image):
18
  return model(image)
19
 
20
+ def make_video(video_path, outdir='./vis_video_depth',encoder='vitb'):
21
  # DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
22
  # model = DepthAnything.from_pretrained('LiheYoung/depth_anything_vitl14').to(DEVICE).eval()
23
  # Define path for temporary processed frames
 
165
  example_files = os.listdir('assets/examples_video')
166
  example_files.sort()
167
  example_files = [os.path.join('assets/examples_video', filename) for filename in example_files]
168
+ examples = gr.Examples(examples=example_files, inputs=input_video, outputs=processed_video, fn=on_submit, cache_examples=True)
169
 
170
 
171
  if __name__ == '__main__':