quazarcom
/

object_detection_v3

Model card Files Files and versions

quazarcom commited on Apr 10, 2025

Commit

231035d

·

verified ·

1 Parent(s): 094555a

Update app.py

Files changed (1) hide show

app.py +1 -1

app.py CHANGED Viewed

@@ -22,7 +22,7 @@ class VisionTransformer(nn.Module):
         return self.model(x)
 # ==== Load the trained weights ====
-model_path = "./models/your_model.pth"
 device = torch.device("cpu")  # MPS not available on HF Spaces
 model = VisionTransformer(num_classes=2, model_name="vit_small_patch16_224")

         return self.model(x)
 # ==== Load the trained weights ====
+model_path = "./models/vit_small_patch16_224_best.pth"
 device = torch.device("cpu")  # MPS not available on HF Spaces
 model = VisionTransformer(num_classes=2, model_name="vit_small_patch16_224")