dd

app.py

@@ -22,17 +22,10 @@ def greet(url):
     masks_queries_logits = outputs.masks_queries_logits
 
     # you can pass them to processor for postprocessing
-    predicted_semantic_map = processor.post_process_semantic_segmentation(outputs, target_sizes=[url.size])[0]
+    predicted_semantic_map = processor.post_process_semantic_segmentation(outputs, target_sizes=[image.size])[0]
+    predicted_semantic_map = predicted_semantic_map.numpy()
 
-    sepia_filter = np.array([
-        [0.393, 0.769, 0.189],
-        [0.349, 0.686, 0.168],
-        [0.272, 0.534, 0.131]
-    ])
-    sepia_img = predicted_semantic_map.dot(sepia_filter.T)
-    sepia_img /= sepia_img.max()
-
-    return sepia_img
+    return predicted_semantic_map
 
 
 url = "http://images.cocodataset.org/val2017/000000039769.jpg"