Spaces:
Runtime error
Runtime error
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"
|