unfinished

app.py

@@ -14,4 +14,4 @@ tm_config = timm.data.resolve_model_data_config(model)
 tm_trans = timm.data.create_transform(**tm_config, is_training=False)
 
 def launch(img):
-    output = model(transforms(img).unsqueeze(0))
+    tm_output = nsfw_tm(transforms(img).unsqueeze(0))