Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -35,7 +35,9 @@ text_prompt="Transform this image into a work of art by changing its style and c
|
|
35 |
import os
|
36 |
os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "expandable_segments:True"
|
37 |
#pipe.to('cuda')
|
38 |
-
|
|
|
|
|
39 |
|
40 |
sam_checkpoint = "sam_vit_h_4b8939.pth"
|
41 |
model_type = "vit_h"
|
|
|
35 |
import os
|
36 |
os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "expandable_segments:True"
|
37 |
#pipe.to('cuda')
|
38 |
+
import sys
|
39 |
+
sys.path.append("..")
|
40 |
+
from segment_anything import sam_model_registry, SamPredictor
|
41 |
|
42 |
sam_checkpoint = "sam_vit_h_4b8939.pth"
|
43 |
model_type = "vit_h"
|