Update app.py
Browse files
app.py
CHANGED
|
@@ -137,7 +137,7 @@ class GatedHead(torch.nn.Module):
|
|
| 137 |
|
| 138 |
model.head = GatedHead(min(model.head.weight.shape), 9083)
|
| 139 |
|
| 140 |
-
safetensors.torch.load_model(model, "
|
| 141 |
model.eval()
|
| 142 |
|
| 143 |
with open("tagger_tags.json", "r") as file:
|
|
|
|
| 137 |
|
| 138 |
model.head = GatedHead(min(model.head.weight.shape), 9083)
|
| 139 |
|
| 140 |
+
safetensors.torch.load_model(model, "JTP_PILOT2-e3-vit_so400m_patch14_siglip_384.safetensors")
|
| 141 |
model.eval()
|
| 142 |
|
| 143 |
with open("tagger_tags.json", "r") as file:
|