Merge pull request replicate#69 from replicate/better-readability

hard to read conditional
chenxwh · Jan 6, 2023 · 3851052 · 3851052
2 parents 05496d2 + 5fad8d9
commit 3851052
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/predict.py b/predict.py
@@ -118,10 +118,12 @@ def predict(
 
         output_paths = []
         for i, sample in enumerate(output.images):
-            if output.nsfw_content_detected and not output.nsfw_content_detected[i]:
-                output_path = f"/tmp/out-{i}.png"
-                sample.save(output_path)
-                output_paths.append(Path(output_path))
+            if output.nsfw_content_detected and output.nsfw_content_detected[i]:
+                continue
+
+            output_path = f"/tmp/out-{i}.png"
+            sample.save(output_path)
+            output_paths.append(Path(output_path))
 
         if len(output_paths) == 0:
             raise Exception(