SahilCarterr commited on
Commit
120b17e
·
verified ·
1 Parent(s): 1f279e9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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
- from segment-anything import sam_model_registry, SamPredictor
 
 
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"