Update app.py

app.py

@@ -6,7 +6,7 @@ import numpy as np
 from transformers import SamModel, SamProcessor
 
 
-model = SamModel.from_pretrained('facebook/sam-vit-huge').to('cuda')
+model = SamModel.from_pretrained('facebook/sam-vit-huge').to('cpu')
 processor = SamProcessor.from_pretrained('facebook/sam-vit-huge')
 
 
@@ -14,7 +14,7 @@ def set_predictor(image):
     """
     Creates a Sam predictor object based on a given image and model.
     """
-    inputs = processor(image, return_tensors='pt').to('cuda')
+    inputs = processor(image, return_tensors='pt').to('cpu')
     image_embedding = model.get_image_embeddings(inputs['pixel_values'])
 
     return [image, image_embedding, 'Done']
@@ -27,7 +27,7 @@ def get_polygon(points, image, image_embedding):
     """
     points = [int(w) for w in points.split(',')]
 
-    inputs = processor(image, input_boxes=[points], return_tensors="pt").to('cuda')
+    inputs = processor(image, input_boxes=[points], return_tensors="pt").to('cpu')
 
     # pop the pixel_values as they are not neded
     inputs.pop("pixel_values", None)