Update app.py

app.py

@@ -4,6 +4,7 @@ import os
 import requests
 import timm
 import torch
+import json
 
 nsfw_tf = pipeline(model="carbon225/vit-base-patch16-224-hentai")
 
@@ -21,14 +22,16 @@ if not os.path.exists("timm.ckpt"):
 else:
     print("Model already exists, skipping redownload")
 
+with open("timmcfg.json") as file:
+    tm_cfg = json.load(file)
 
 nsfw_tm = timm.create_model(
     "caformer_s36.sail_in22k_ft_in1k_384",
     checkpoint_path="./timm.ckpt",
-    pretrained_cfg="./timmcfg.json",
+    model_config=tm_cfg,
     pretrained=True
 ).eval()
-tm_config = timm.data.resolve_model_data_config(nsfw_tm.pretrained_cfg, model=nsfw_tm)
+tm_config = timm.data.resolve_model_data_config(nsfw_tm)
 tm_trans = timm.data.create_transform(**tm_config)