Spaces:

Thouph
/

Furrence-2-Large-Demo

Running

Thouph commited on May 7

Commit

1d4a2c7

verified ·

1 Parent(s): 886a2e7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -226,7 +226,7 @@ tagger_model = timm.create_model(
     num_classes=9083,
 )  # type: VisionTransformer
 tagger_model.head = GatedHead(min(tagger_model.head.weight.shape), 9083)
-safetensors.torch.load_model(tagger_model, "JTP_PILOT2-2-e3-vit_so400m_patch14_siglip_384.safetensors")
 tagger_model.eval()

     num_classes=9083,
 )  # type: VisionTransformer
 tagger_model.head = GatedHead(min(tagger_model.head.weight.shape), 9083)
+safetensors.torch.load_model(tagger_model, "JTP_PILOT2-e3-vit_so400m_patch14_siglip_384.safetensors")
 tagger_model.eval()