Spaces:
Sleeping
Sleeping
config download access
Browse files- app.py +1 -1
- inference_sam.py +1 -2
app.py
CHANGED
|
@@ -15,7 +15,7 @@ dotenv.load_dotenv()
|
|
| 15 |
import numpy as np
|
| 16 |
import gradio as gr
|
| 17 |
import glob
|
| 18 |
-
from inference_sam import segmentation_sam
|
| 19 |
from explanations import explain
|
| 20 |
from inference_resnet import get_triplet_model
|
| 21 |
from inference_beit import get_triplet_model_beit
|
|
|
|
| 15 |
import numpy as np
|
| 16 |
import gradio as gr
|
| 17 |
import glob
|
| 18 |
+
# from inference_sam import segmentation_sam
|
| 19 |
from explanations import explain
|
| 20 |
from inference_resnet import get_triplet_model
|
| 21 |
from inference_beit import get_triplet_model_beit
|
inference_sam.py
CHANGED
|
@@ -31,8 +31,7 @@ if not os.path.exists('model'):
|
|
| 31 |
print(f"Read token:{token}")
|
| 32 |
if token is None:
|
| 33 |
print("warning! A read token in env variables is needed for authentication.")
|
| 34 |
-
|
| 35 |
-
snapshot_download(repo_id=REPO_ID, token=token,repo_type='model',local_dir='model')
|
| 36 |
|
| 37 |
model_path = os.path.join('model', 'sam_02-06_dice_mse_0.pth')
|
| 38 |
sam = sam_model_registry["default"](model_path)
|
|
|
|
| 31 |
print(f"Read token:{token}")
|
| 32 |
if token is None:
|
| 33 |
print("warning! A read token in env variables is needed for authentication.")
|
| 34 |
+
snapshot_download(repo_id=REPO_ID, token=token,repo_type='model',local_dir='model')
|
|
|
|
| 35 |
|
| 36 |
model_path = os.path.join('model', 'sam_02-06_dice_mse_0.pth')
|
| 37 |
sam = sam_model_registry["default"](model_path)
|