Update app.py

app.py

@@ -35,7 +35,14 @@ text_prompt="Transform this image into a work of art by changing its style and c
 import os
 os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "expandable_segments:True"
 #pipe.to('cuda')
+from segment_anything import sam_model_registry, SamPredictor
 
+sam_checkpoint = "sam_vit_h_4b8939.pth"
+model_type = "vit_h"
+device = "cpu"
+
+sam = sam_model_registry[model_type](checkpoint=sam_checkpoint)
+sam.to(device=device)
 
 def image_enhance(image, mask):
   image = np.array(image)