Spaces:

spuun
/

nsfw-det

Sleeping

spuun commited on Jun 25, 2023

Commit

74249eb

•

1 Parent(s): ae4e131

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -48,7 +48,7 @@ def launch(img):
     tm_image = load_image(img, mode='RGB')
     tm_input_ = _img_encode(tm_image, size=(256, 256))[None, ...]
     tm_output, = nsfw_tm.run(['output'], {'input': tm_input_})
-    tm_output = zip(tm_cfg["labels"], map(lambda x: x.item(), tm_output[0]))[0][0]
     match tm_output:
         case "safe":

     tm_image = load_image(img, mode='RGB')
     tm_input_ = _img_encode(tm_image, size=(256, 256))[None, ...]
     tm_output, = nsfw_tm.run(['output'], {'input': tm_input_})
+    tm_output = list(zip(tm_cfg["labels"], map(lambda x: x.item(), tm_output[0])))[0][0]
     match tm_output:
         case "safe":