clean imports, strip underscores from tags

app.py

@@ -1,21 +1,16 @@
 import json
 
-from PIL import Image
 import gradio as gr
+from PIL import Image
+import safetensors.torch
+import spaces
+import timm
+from timm.models import VisionTransformer
 import torch
 from torchvision.transforms import transforms
 from torchvision.transforms import InterpolationMode
 import torchvision.transforms.functional as TF
 
-import spaces
-
-import huggingface_hub
-import timm
-from timm.models import VisionTransformer
-import safetensors.torch
-
-
-torch.jit.script = lambda f: f
 torch.set_grad_enabled(False)
 
 class Fit(torch.nn.Module):
@@ -130,6 +125,9 @@ with open("tagger_tags.json", "r") as file:
     tags = json.load(file) # type: dict
 allowed_tags = list(tags.keys())
 
+for idx, tag in enumerate(allowed_tags):
+    allowed_tags[idx] = tag.replace("_", " ")
+
 @spaces.GPU(duration=5)
 def create_tags(image, threshold):
     img = image.convert('RGB')