Update tasks/image.py

tasks/image.py

@@ -25,7 +25,7 @@ ROUTE = "/image"
 
 device = "cpu"
 model_path = "mobilevit_segmentation_full_data.pth"
-feature_extractor = MobileViTImageProcessor.from_pretrained("apple/deeplabv3-mobilevit-xx-small").to(device)
+feature_extractor = MobileViTImageProcessor.from_pretrained("apple/deeplabv3-mobilevit-xx-small")
 model = MobileViTForSemanticSegmentation.from_pretrained("apple/deeplabv3-mobilevit-xx-small").to(device)
 model.load_state_dict(torch.load(model_path), map_location=torch.device('cpu'))
 model.eval()