Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -14,7 +14,7 @@ from PIL import Image
|
|
14 |
from safetensors.torch import load_file
|
15 |
from torchvision.transforms import ToPILImage, ToTensor
|
16 |
from transformers import AutoModelForImageSegmentation
|
17 |
-
|
18 |
|
19 |
huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
|
20 |
|
|
|
14 |
from safetensors.torch import load_file
|
15 |
from torchvision.transforms import ToPILImage, ToTensor
|
16 |
from transformers import AutoModelForImageSegmentation
|
17 |
+
from utils import extract_object, get_model_from_config, resize_and_center_crop
|
18 |
|
19 |
huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
|
20 |
|