Update app.py

app.py

@@ -28,7 +28,8 @@ with open("timmcfg.json") as file:
 nsfw_tm = timm.create_model(
     "caformer_s36.sail_in22k_ft_in1k_384",
     checkpoint_path="./timm.ckpt",
-    model_config=tm_cfg
+    model_config=tm_cfg,
+    num_classes=3
 ).eval()
 tm_config = timm.data.resolve_model_data_config(nsfw_tm)
 tm_trans = timm.data.create_transform((256, 256), **tm_config, is_training=False)