amaye15
commited on
Commit
·
c8a6ab0
1
Parent(s):
e34d5e8
Forgot login
Browse files
app.py
CHANGED
|
@@ -5,7 +5,7 @@ import numpy as np
|
|
| 5 |
from sam2.sam2_image_predictor import SAM2ImagePredictor
|
| 6 |
from uuid import uuid4
|
| 7 |
import os
|
| 8 |
-
from huggingface_hub import upload_folder
|
| 9 |
from PIL import Image as PILImage
|
| 10 |
from datasets import Dataset, Features, Array2D, Image
|
| 11 |
import shutil
|
|
@@ -18,7 +18,7 @@ PREDICTOR = SAM2ImagePredictor.from_pretrained(MODEL, device=DEVICE)
|
|
| 18 |
|
| 19 |
DESTINATION_DS = "amaye15/object-segmentation"
|
| 20 |
|
| 21 |
-
|
| 22 |
|
| 23 |
IMAGE = None
|
| 24 |
MASKS = None
|
|
|
|
| 5 |
from sam2.sam2_image_predictor import SAM2ImagePredictor
|
| 6 |
from uuid import uuid4
|
| 7 |
import os
|
| 8 |
+
from huggingface_hub import upload_folder, login
|
| 9 |
from PIL import Image as PILImage
|
| 10 |
from datasets import Dataset, Features, Array2D, Image
|
| 11 |
import shutil
|
|
|
|
| 18 |
|
| 19 |
DESTINATION_DS = "amaye15/object-segmentation"
|
| 20 |
|
| 21 |
+
login(os.getenv("TOKEN"))
|
| 22 |
|
| 23 |
IMAGE = None
|
| 24 |
MASKS = None
|