Instructions to use RedRocket/JointTaggerProject with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- timm
How to use RedRocket/JointTaggerProject with timm:
import timm model = timm.create_model("hf_hub:RedRocket/JointTaggerProject", pretrained=True) - Notebooks
- Google Colab
- Kaggle
Update JTP_PILOT/inference_gradio.py
Browse files
JTP_PILOT/inference_gradio.py
CHANGED
|
@@ -126,7 +126,7 @@ if torch.cuda.is_available():
|
|
| 126 |
|
| 127 |
model.eval()
|
| 128 |
|
| 129 |
-
with open("
|
| 130 |
tags = json.load(file) # type: dict
|
| 131 |
allowed_tags = list(tags.keys())
|
| 132 |
|
|
|
|
| 126 |
|
| 127 |
model.eval()
|
| 128 |
|
| 129 |
+
with open("tags.json", "r") as file:
|
| 130 |
tags = json.load(file) # type: dict
|
| 131 |
allowed_tags = list(tags.keys())
|
| 132 |
|