init project
Browse files- app.py +4 -2
- requirements.txt +9 -1
app.py
CHANGED
|
@@ -8,7 +8,10 @@ import os
|
|
| 8 |
import sys
|
| 9 |
sys.path.append(os.path.abspath('./modules'))
|
| 10 |
|
| 11 |
-
|
|
|
|
|
|
|
|
|
|
| 12 |
|
| 13 |
import math
|
| 14 |
import tempfile
|
|
@@ -45,7 +48,6 @@ sys.path.append(os.path.abspath('./modules/ultralytics'))
|
|
| 45 |
|
| 46 |
from transformers import AutoTokenizer, AutoModel, AutoProcessor, SamModel
|
| 47 |
# from modules.mast3r.model import AsymmetricMASt3R
|
| 48 |
-
from mast3r.model import AsymmetricMASt3R
|
| 49 |
|
| 50 |
# from modules.sam2.build_sam import build_sam2_video_predictor
|
| 51 |
from modules.mobilesamv2.promt_mobilesamv2 import ObjectAwareModel
|
|
|
|
| 8 |
import sys
|
| 9 |
sys.path.append(os.path.abspath('./modules'))
|
| 10 |
|
| 11 |
+
HERE_PATH = os.path.normpath(os.path.dirname(__file__)) # noqa
|
| 12 |
+
MASt3R_REPO_PATH = os.path.normpath(os.path.join(HERE_PATH, './mast3r')) # noqa
|
| 13 |
+
sys.path.insert(0, MASt3R_REPO_PATH) # noqa
|
| 14 |
+
from mast3r.model import AsymmetricMASt3R
|
| 15 |
|
| 16 |
import math
|
| 17 |
import tempfile
|
|
|
|
| 48 |
|
| 49 |
from transformers import AutoTokenizer, AutoModel, AutoProcessor, SamModel
|
| 50 |
# from modules.mast3r.model import AsymmetricMASt3R
|
|
|
|
| 51 |
|
| 52 |
# from modules.sam2.build_sam import build_sam2_video_predictor
|
| 53 |
from modules.mobilesamv2.promt_mobilesamv2 import ObjectAwareModel
|
requirements.txt
CHANGED
|
@@ -10,4 +10,12 @@ matplotlib
|
|
| 10 |
scipy
|
| 11 |
roma
|
| 12 |
hydra-core
|
| 13 |
-
iopath
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 10 |
scipy
|
| 11 |
roma
|
| 12 |
hydra-core
|
| 13 |
+
iopath
|
| 14 |
+
numpy<2.0.0
|
| 15 |
+
scikit-learn
|
| 16 |
+
pillow-heif
|
| 17 |
+
gradio
|
| 18 |
+
tqdm
|
| 19 |
+
tensorboard
|
| 20 |
+
pyglet<2
|
| 21 |
+
huggingface-hub[torch]>=0.22
|