Update app.py

app.py

@@ -8,7 +8,7 @@ from itertools import chain
 from transformers import SamModel, SamProcessor
 
 
-model = SamModel.from_pretrained('facebook/sam-vit-huge').to('cuda')
+model = SamModel.from_pretrained('facebook/sam-vit-huge')
 processor = SamProcessor.from_pretrained('facebook/sam-vit-huge')
 
 
@@ -16,7 +16,7 @@ def set_predictor(image):
     """
     Creates a Sam predictor object based on a given image and model.
     """
-    device = 'cuda'
+    device = 'cpu'
     inputs = processor(image, return_tensors='pt').to(device)
     image_embedding = model.get_image_embeddings(inputs['pixel_values'])
 
@@ -28,11 +28,9 @@ def get_polygon(points, image, image_embedding):
     Returns the points of the polygon given a bounding box and a prediction
     made by Sam.
     """
-    #points = [int(w) for w in points.split(',')]
     points = list(chain.from_iterable(points))
-    print(points)
 
-    device = 'cuda'
+    device = 'cpu'
     inputs = processor(image, input_boxes=[points], return_tensors="pt").to(device)
 
     # pop the pixel_values as they are not neded