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_PILOT2/inference_gradio.py
Browse files
JTP_PILOT2/inference_gradio.py
CHANGED
|
@@ -145,7 +145,7 @@ if torch.cuda.is_available():
|
|
| 145 |
|
| 146 |
model.eval()
|
| 147 |
|
| 148 |
-
with open("
|
| 149 |
tags = json.load(file) # type: dict
|
| 150 |
allowed_tags = list(tags.keys())
|
| 151 |
|
|
|
|
| 145 |
|
| 146 |
model.eval()
|
| 147 |
|
| 148 |
+
with open("tags.json", "r") as file:
|
| 149 |
tags = json.load(file) # type: dict
|
| 150 |
allowed_tags = list(tags.keys())
|
| 151 |
|