brandonsmart commited on
Commit
7dfcb4e
1 Parent(s): ccd5a59

Supporting ZeroGPU

Browse files
Files changed (2) hide show
  1. .gitignore +1 -0
  2. demo.py +2 -1
.gitignore ADDED
@@ -0,0 +1 @@
 
 
1
+ .DS_Store
demo.py CHANGED
@@ -10,6 +10,7 @@ import os
10
  import sys
11
  import tempfile
12
 
 
13
  import gradio
14
  import torch
15
  from huggingface_hub import hf_hub_download
@@ -22,7 +23,7 @@ from mast3r.utils.misc import hash_md5
22
  import main
23
  import utils.export as export
24
 
25
-
26
  def get_reconstructed_scene(outdir, model, device, silent, image_size, ios_mode, filelist):
27
 
28
  assert len(filelist) == 1 or len(filelist) == 2, "Please provide one or two images"
 
10
  import sys
11
  import tempfile
12
 
13
+ import spaces
14
  import gradio
15
  import torch
16
  from huggingface_hub import hf_hub_download
 
23
  import main
24
  import utils.export as export
25
 
26
+ @spaces.GPU(duration=10)
27
  def get_reconstructed_scene(outdir, model, device, silent, image_size, ios_mode, filelist):
28
 
29
  assert len(filelist) == 1 or len(filelist) == 2, "Please provide one or two images"