Spaces:
Running
on
Zero
Running
on
Zero
brandonsmart
commited on
Commit
•
7dfcb4e
1
Parent(s):
ccd5a59
Supporting ZeroGPU
Browse files- .gitignore +1 -0
- 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"
|